diff --git a/Biblioteka_DL/dllib-mlflow.py b/Biblioteka_DL/dllib-mlflow.py
index dc5043e..2be6024 100644
--- a/Biblioteka_DL/dllib-mlflow.py
+++ b/Biblioteka_DL/dllib-mlflow.py
@@ -289,7 +289,7 @@ def my_main(epochs):
     loss_fn = nn.CrossEntropyLoss()
     # epochs = 1000
     # epochs = epochs
-    alpha = int(sys.argv[1]) if len(sys.argv) > 1 else 20
+    epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 20
     mlflow.log_param("epochs", epochs)
     
     def print_(loss):