diff --git a/laborious/activities/gates.py b/laborious/activities/gates.py
index 7acafd3..0b747f9 100644
--- a/laborious/activities/gates.py
+++ b/laborious/activities/gates.py
@@ -479,7 +479,7 @@ class Gates(MinioManager):
             minio_repo=self.minio_repository,
             model_name=input_data['model_name'],
             operation='transform',
-            workflow_metadata=metadata,
+            workflow_metadata=metadata
         )
 
     @activity.defn(name='format_prediction')
diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py
index 27d1b92..e6c75e1 100644
--- a/laborious/activities/mlflow.py
+++ b/laborious/activities/mlflow.py
@@ -1,6 +1,6 @@
 from temporalio import activity, workflow
 
-from laborious.utils.repository.minio_manager import MinioManager
+
 
 with workflow.unsafe.imports_passed_through():
     import traceback
@@ -20,7 +20,7 @@ with workflow.unsafe.imports_passed_through():
         now,
     )
     from sientia_do.utils.formatters import create_sample_dict
-
+    from laborious.utils.repository.minio_manager import MinioManager
     from laborious.utils.models.minio_dataframe_payload import MinioDataFramePayload
     from laborious.utils.repository.model_repository import MLFlowRepository