diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py
index 28086d5..b96cc60 100644
--- a/laborious/activities/mlflow.py
+++ b/laborious/activities/mlflow.py
@@ -60,7 +60,7 @@ class MLFlow(BaseActivity):
         self.mlflow_password = mlflow_password
 
         self.model_monitoring_repository = MLFlowRepository(
-            f"{mlflow_host}:{mlflow_port}", mlflow_username, mlflow_password, logger.base_logger
+            f"{mlflow_host}:{mlflow_port}", mlflow_username, mlflow_password, logger
         )
 
     @activity.defn(name="request_transform")