diff --git a/tests/integrations/test_mlflow.py b/tests/integrations/test_mlflow.py
index ec133c5e9..f57554692 100644
--- a/tests/integrations/test_mlflow.py
+++ b/tests/integrations/test_mlflow.py
@@ -1,4 +1,3 @@
-import pytest
 from unittest.mock import Mock, patch
 import sys
 
@@ -46,9 +45,7 @@ class TestMLflowIntegration:
         with patch('crewai.integrations.mlflow.mlflow') as mock_mlflow:
             mock_mlflow.tracing.start_span.return_value = Mock()
             
-            import mlflow
-            mlflow.crewai.autolog()
-            
+            autolog()
             assert mlflow_listener._autolog_enabled is True
             
             from crewai import Agent, Task, Crew