diff --git a/pymolint/test_base.py b/pymolint/test_base.py
index 3ce9288..5ff1c37 100644
--- a/pymolint/test_base.py
+++ b/pymolint/test_base.py
@@ -1,6 +1,6 @@
 import numpy as np 
 from itertools import permutations
-from base import *
+from pymolint.base import *
 
 
 def test_base():