diff --git a/s487175-create-dataset-script.py b/s487175-create-dataset-script.py
index 93e297c..055700a 100644
--- a/s487175-create-dataset-script.py
+++ b/s487175-create-dataset-script.py
@@ -6,10 +6,9 @@ import pandas as pd
 diamonds = pd.read_csv('diamonds.csv')
 
 #Wyświetlenie zbioru danych
-cutoff = os.environ.get('CUTOFF')
+cutoff = os.environ.get('CUTOFF', None)
 if cutoff is None:
     print("CUTOFF parameter not set.")
-    diamonds
 else:
     cutoff = int(cutoff)
     diamonds = diamonds[:cutoff]