diff --git a/tests/test_ext.py b/tests/test_ext.py
index 57b5ac4..373b4bc 100644
--- a/tests/test_ext.py
+++ b/tests/test_ext.py
@@ -2,7 +2,7 @@
 
 from pyramid.path import DottedNameResolver
 
-from pyramid_jinja2.tests.base import Base
+from tests.base import Base
 
 
 class TestExtensions(Base, unittest.TestCase):
@@ -16,7 +16,7 @@ def test_custom_extension(self):
         settings = parse_env_options_from_settings(options, "", maybe_dotted, None)
         env = create_environment_from_options(settings, {})
         ext = env.extensions["tests.extensions.TestExtension"]
-        from pyramid_jinja2.tests import extensions
+        from tests import extensions
 
         self.assertEqual(ext.__class__, extensions.TestExtension)
 
diff --git a/tests/test_it.py b/tests/test_it.py
index 369a675..3eac940 100644
--- a/tests/test_it.py
+++ b/tests/test_it.py
@@ -3,7 +3,7 @@
 
 from pyramid import testing
 
-from pyramid_jinja2.tests.base import Base, Mock
+from tests.base import Base, Mock
 
 
 def dummy_filter(value):