diff --git a/create_dataset.py b/create_dataset.py
index 068a6e4..ecf7aef 100644
--- a/create_dataset.py
+++ b/create_dataset.py
@@ -49,7 +49,7 @@ water_test, water_dev = train_test_split(water_test, train_size=0.66, random_sta
 
 #water["is_safe"].value_counts().plot(kind="bar")
 
-
+import os
 
 directory ='dane'
 if not os.path.exists(directory):