diff --git a/test.sh b/test.sh
index f945071..1bc3518 100644
--- a/test.sh
+++ b/test.sh
@@ -1,6 +1,6 @@
 curl -OL https://git.wmi.amu.edu.pl/s434788/ium_434788/raw/branch/master/winequality-red.csv
 
-head -n 1 wine.csv >header.csv
-tail -n +2 wine.csv | shuf > data.shuffled
+head -n 1 winequality-red.csv > header.csv
+tail -n +2 winequality-red.csv | shuf > data.shuffled
 head -n 1000 data.shuffled > train.data.csv
 cat header.tsv train.data.csv > train.csv