diff --git a/tests/unit_test.py b/tests/unit_test.py
index eeb8269a2..28b44b41a 100644
--- a/tests/unit_test.py
+++ b/tests/unit_test.py
@@ -66,7 +66,7 @@
     export_env,
 )
 
-from tests.unit_test_support import (
+from unit_test_support import (
     createClusterWithConfig,
     createTestDDP,
     createDDPJob_no_cluster,