diff --git a/download_dataset.sh b/download_dataset.sh
index 64ac77a..71bad14 100644
--- a/download_dataset.sh
+++ b/download_dataset.sh
@@ -5,9 +5,9 @@ kaggle datasets download -d nikhil7280/student-performance-multiple-linear-regre
 
 # change dataset name to data.csv
 mv Student_Performance.csv data.csv
-
+echo "$2"
 # cut off rows
-head -n $2 data.csv > data.csv.tmp && mv data.csv.tmp data.csv
+head -n "$2" data.csv > data.csv.tmp && mv data.csv.tmp data.csv
 
 # get data size
 data_size=$(wc -l < data.csv)