diff --git a/tests/unit/test_test_dag.py b/tests/unit/test_test_dag.py
index a1d515b..cf2abbb 100644
--- a/tests/unit/test_test_dag.py
+++ b/tests/unit/test_test_dag.py
@@ -1,4 +1,4 @@
-from dags.example_dag.example_dag import fetch, pull
+from example_dag.example_dag import fetch, pull
 
 
 def test_test_dag_fetch():