diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py
index 4e5e3a6..af53a22 100644
--- a/laborious/activities/mlflow.py
+++ b/laborious/activities/mlflow.py
@@ -23,7 +23,7 @@ class MLFlow(BaseActivity):
         self.mlflow_password = mlflow_password
 
         self.model_monitoring_repository = MLFlowRepository(
-            f"{mlflow_host}:{mlflow_port}", mlflow_username, mlflow_password
+            f"{mlflow_host}:{mlflow_port}", mlflow_username, mlflow_password, logger.base_logger
         )
 
     @activity.defn(name="request_transform")
diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py
index c3fe550..d518750 100644
--- a/laborious/utils/repository/model_repository.py
+++ b/laborious/utils/repository/model_repository.py
@@ -19,10 +19,11 @@ from sientia.ModelServing import ModelServing
 
 
 class MLFlowRepository():
-    def __init__(self, host, username, password):
+    def __init__(self, host, username, password, logger):
 
         self.model_serving = ModelServing(tracking_uri=host,
-                                          username=username, password=password)
+                                          username=username, password=password,
+                                          logger=logger)
 
     def transform(self, model_name: str, data: pd.DataFrame, model_retention: int):
         """