diff --git a/Zadanie_08_and_09_MLflow.py b/Zadanie_08_and_09_MLflow.py
index d57db95..4637617 100644
--- a/Zadanie_08_and_09_MLflow.py
+++ b/Zadanie_08_and_09_MLflow.py
@@ -17,6 +17,7 @@ with mlflow.start_run():
     batch_param = int(sys.argv[1]) if len(sys.argv) > 1 else 16
     epoch_param = int(sys.argv[2]) if len(sys.argv) > 2 else 15
 
+    mlflow.set_tracking_uri("http://172.17.0.1:5000")
     mlflow.set_experiment("s434788")
     mlflow.log_param("batch_size", batch_param)
     mlflow.log_param("epochs", epoch_param)