diff --git a/tests/test_hive.py b/tests/test_hive.py
index d85ee9f..7f4b31a 100644
--- a/tests/test_hive.py
+++ b/tests/test_hive.py
@@ -1,7 +1,7 @@
 import unittest
 from mock import MagicMock
 
-import hive
+from policytool import hive
 
 
 class _CursorMock:
diff --git a/tests/test_policyutil.py b/tests/test_policyutil.py
index d632d58..6ca3f8c 100644
--- a/tests/test_policyutil.py
+++ b/tests/test_policyutil.py
@@ -1,6 +1,6 @@
 import unittest
 
-import policyutil
+from policytool import policyutil
 
 
 class TestValidatePolicy(unittest.TestCase):