diff --git a/create_dataset.py b/create_dataset.py
index ecf7aef..d6d65b2 100644
--- a/create_dataset.py
+++ b/create_dataset.py
@@ -56,7 +56,7 @@ if not os.path.exists(directory):
     os.makedirs('dane')
 
 waterNorm.to_csv('waterQuality.csv', index=False)
-water_train.to_csv('dane/diamonds_train.csv', index=False)
-water_test.to_csv('dane/diamonds_test.csv', index=False)
-water_dev.to_csv('dane/diamonds_dev.csv', index=False)
+water_train.to_csv('dane/water_train.csv', index=False)
+water_test.to_csv('dane/water_test.csv', index=False)
+water_dev.to_csv('dane/water_dev.csv', index=False)