diff --git a/simple_regression_lab7.py b/simple_regression_lab7.py
index a2cb60d..2703b8f 100644
--- a/simple_regression_lab7.py
+++ b/simple_regression_lab7.py
@@ -74,7 +74,7 @@ def train(epochs, units, learning_rate, _run):
 
     # Zapis modelu do pliku standardowo poprzez metodę kerasa i poprzez metodę obiektu Experiment z Sacred
     model.save('lego_reg_model')
-    ex.add_artifact('lego_reg_model')
+    ex.add_artifact('lego_reg_model/saved_model.pb')
 
     # Przykładowo zwracamy coś w charakterze wyników, żeby było widoczne w plikach zapisanych przez obserwator
     hist = pd.DataFrame(history.history)