diff --git a/download_dataset.sh b/download_dataset.sh
index 3463cb9..5ea195b 100644
--- a/download_dataset.sh
+++ b/download_dataset.sh
@@ -16,8 +16,8 @@ shuf creditcard.csv > creditcard_shuf.csv
 rm creditcard.csv
 
 # Split the dataset into training and testing
-head -n 10000 creditcard_shuf.csv > creditcard_train.csv
-tail -n +10001 creditcard_shuf.csv > creditcard_test.csv
+tail -n +10001 creditcard_shuf.csv > creditcard_train.csv
+head -n 10000 creditcard_shuf.csv > creditcard_test.csv
 
 # Create a directory for the data
 mkdir -p data