diff --git a/simple_regression_lab7.py b/simple_regression_lab7.py
index eac89bb..b01a8be 100644
--- a/simple_regression_lab7.py
+++ b/simple_regression_lab7.py
@@ -20,14 +20,14 @@ ex.observers.append(FileStorageObserver('runs'))
 # Przykładowa modyfikowalna z Sacred konfiguracja wybranych parametrów treningu
 @ex.config
 def config():
+    epochs = 100
     units = 1
     learning_rate = 0.1
-    epochs = 100
 
 
 # Reszta kodu wrzucona do udekorowanej funkcji train do wywołania przez Sacred, żeby coś było capture'owane
 @ex.capture
-def train(units, learning_rate, _run):
+def train(epochs, units, learning_rate, _run):
 
     # Wczytanie danych
     data_train = pd.read_csv('lego_sets_clean_train.csv')