diff --git a/conftest.py b/conftest.py
index 734f46eed..54ca45107 100644
--- a/conftest.py
+++ b/conftest.py
@@ -271,12 +271,10 @@ def mock_submit_fn(metric, start, end):
     mock_submit = mocker.Mock()
     mock_submit.side_effect = mock_submit_fn
 
-    import helpers
+    from helpers.checkpoint_logger import CheckpointLogger
 
     return mocker.patch.object(
-        helpers.checkpoint_logger.CheckpointLogger,
+        CheckpointLogger,
         "submit_subflow",
         mock_submit,
     )
-
-    return mock_submit