diff --git a/tests/test_collect_data.py b/tests/test_collect_data.py
index be7d192d..95b7e4c9 100644
--- a/tests/test_collect_data.py
+++ b/tests/test_collect_data.py
@@ -44,14 +44,14 @@
 except ModuleNotFoundError:
     # case of upload everything to argo, no context needed
     pass
-from .context import (
+from context import (
     default_host,
     default_image,
     skip_ut_with_dflow,
     skip_ut_with_dflow_reason,
     upload_python_packages,
 )
-from .mocked_ops import (
+from mocked_ops import (
     MockedCollectData,
 )