diff --git a/lab5/create/create_dataset.py b/lab5/create/create_dataset.py
index 45d451c..e2eae99 100644
--- a/lab5/create/create_dataset.py
+++ b/lab5/create/create_dataset.py
@@ -30,7 +30,7 @@ train = pd.concat([X_train, Y_train], axis=1)
 test = pd.concat([X_test, Y_test], axis=1)
 valid = pd.concat([X_valid, Y_valid], axis=1)
 
-train.to_csv('train.cs', index_col = False)
-test.to_csv('test.csv', index_col = False)
-valid.to_csv('valid.csv', index_col = False)
+train.to_csv('train.cs', index=False)
+test.to_csv('test.csv', index=False)
+valid.to_csv('valid.csv', index=False)