diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/liquidation_test.py b/tests/liquidation_test.py
index 8bc43f1..0771c2b 100644
--- a/tests/liquidation_test.py
+++ b/tests/liquidation_test.py
@@ -1,6 +1,7 @@
 import unittest
-import inspector_compound
-import inspector_aave
+
+from mev_inspect import inspector_compound
+from mev_inspect import inspector_aave
 
 class TestLiquidations (unittest.TestCase):
     def test_compound_liquidation(self):
@@ -16,4 +17,4 @@ class TestLiquidations (unittest.TestCase):
 
 
 if __name__ == '__main__':
-    unittest.main()
\ No newline at end of file
+    unittest.main()
diff --git a/tests/tokenflow_test.py b/tests/tokenflow_test.py
index 529bb35..7a13ba1 100644
--- a/tests/tokenflow_test.py
+++ b/tests/tokenflow_test.py
@@ -1,5 +1,7 @@
 import unittest
-import tokenflow
+
+from mev_inspect import tokenflow
+
 
 class TestTokenFlow (unittest.TestCase):
     def test_simple_arb(self):
@@ -24,4 +26,4 @@ class TestTokenFlow (unittest.TestCase):
         self.assertEqual(res['dollar_flows'], [0,0])       
 
 if __name__ == '__main__':
-    unittest.main()
\ No newline at end of file
+    unittest.main()