diff --git a/data_download.sh b/data_download.sh
index a5efbfe..050136e 100644
--- a/data_download.sh
+++ b/data_download.sh
@@ -20,7 +20,8 @@ if [[ "$1" == "--cutoff" ]]; then
     echo "Error: cutoff value must be a positive number"
     exit 1
   fi
-  cutoff="$1"
+  cutoff=$(("$1" + 1))
+  echo "CUTOFF $cutoff"
 fi
 
 head -n "$cutoff" train.csv > cutoff_train.csv