diff --git a/DESCRIPTION b/DESCRIPTION
index 83b4455..75bb748 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: TFBlearner
 Title: Functionality for training TF-specific classifiers to predict TF bindings based on ATAC-seq data.
-Version: 0.0.1.1000
+Version: 0.0.1.1001
 Authors@R: 
     person("Emanuel", "Sonder", , "emanuel.sonder@hest.ethz.ch", role = c("aut", "cre"),
            comment = c(ORCID = "0000-0003-4788-9508"))
diff --git a/R/getFeatureMatrix.R b/R/getFeatureMatrix.R
index e94988d..b6f1058 100644
--- a/R/getFeatureMatrix.R
+++ b/R/getFeatureMatrix.R
@@ -239,7 +239,7 @@ getFeatureMatrix <- function(mae,
 
   if(saveHdf5)
   {
-    saveChunk <- fifelse(length(contexts)>3, TRUE, FALSE)
+    saveChunk <- fifelse(length(contexts)>1, TRUE, FALSE)
     if(is.null(outDir)) outDir <- getwd() # .
     if(!is.null(prefix)){
       fileName <- paste(prefix, "feature_matrix", tfName, sep="_")}