diff --git a/process_dataset.py b/process_dataset.py
index 8a93890..2f54b7b 100644
--- a/process_dataset.py
+++ b/process_dataset.py
@@ -18,10 +18,10 @@ print(lego)
 print(lego.describe(include='all'))
 
 # pierwszy podzial, wydzielamy zbior treningowy
-lego_train, lego_rem = train_test_split(lego, train_size=0.8, random_state=1)
+lego_train, lego_rem = train_test_split(lego, train_size=0.8)
 
 # drugi podział, wydzielamy walidacyjny i testowy
-lego_valid, lego_test = train_test_split(lego_rem, test_size=0.5, random_state=1)
+lego_valid, lego_test = train_test_split(lego_rem, test_size=0.5)
 
 # zapis
 lego.to_csv('lego_sets_clean.csv', index=None, header=True)