diff --git a/download.sh b/download.sh
index 1a595f7..1b58264 100755
--- a/download.sh
+++ b/download.sh
@@ -5,9 +5,9 @@ dataset_operation() {
     len2=$(($len1/10))
     len3=$(($len2*2))
     len4=$(($len3+1))
-    head -n $len2 all_games.csv.shuf > all_games.csv.test
-    head -n $len3 all_games.csv.shuf | tail -n $len2 > all_games.csv.dev
-    tail -n +$len4 all_games.csv.shuf > all_games.csv.train
+    head -n $len2 all_games.csv.shuf > all_games_test.csv
+    head -n $len3 all_games.csv.shuf | tail -n $len2 > all_games_dev.csv
+    tail -n +$len4 all_games.csv.shuf > all_games_train.csv
     cp all_games.csv /dane/all_games.csv
     wc -l all_games.csv.*
 }