diff --git a/test/test_essence_extractor.py b/test/test_essence_extractor.py
index 8024611..1eb09e6 100644
--- a/test/test_essence_extractor.py
+++ b/test/test_essence_extractor.py
@@ -18,7 +18,7 @@
 import os
 import collections
 
-from cord_workflow_airflow_extensions.essence_extractor import EssenceExtractor
+from tools.essence_extractor import EssenceExtractor
 
 
 test_path = os.path.abspath(os.path.dirname(os.path.realpath(__file__)))
diff --git a/test/test_kickstarter.py b/test/test_kickstarter.py
index c121a57..3c2cd89 100644
--- a/test/test_kickstarter.py
+++ b/test/test_kickstarter.py
@@ -15,7 +15,7 @@
 from __future__ import absolute_import
 import unittest
 
-from cord_workflow_airflow_extensions.kickstarter import check_web_live
+from tools.kickstarter import check_web_live
 
 
 class TestKickstarter(unittest.TestCase):
diff --git a/test/test_workflow_ctl.py b/test/test_workflow_ctl.py
index 38090d7..c65a09d 100644
--- a/test/test_workflow_ctl.py
+++ b/test/test_workflow_ctl.py
@@ -15,7 +15,7 @@
 from __future__ import absolute_import
 import unittest
 
-from cord_workflow_airflow_extensions.workflow_ctl import register_workflow
+from tools.workflow_ctl import register_workflow
 
 
 class TestWorkflowCtl(unittest.TestCase):