diff --git a/tests/test_internal_controller.py b/tests/test_internal_controller.py
index b36b93f1bd40450666b6d9ebd6b50e32e4c92566..a50911473d38709dc4f194a021709bb4223fb5be 100644
--- a/tests/test_internal_controller.py
+++ b/tests/test_internal_controller.py
@@ -22,7 +22,7 @@ from __future__ import absolute_import
 from flask import json
 
 from aura_engine_api.rest.models.clock_info import ClockInfo  # noqa: E501
-from aura_engine_api.rest.test import BaseTestCase
+from tests import BaseTestCase
 
 
 class TestInternalController(BaseTestCase):
diff --git a/tests/test_public_controller.py b/tests/test_public_controller.py
index 431bb7656bf543ff9b2ee294b3edeb5b8d63183c..4d22c26ba9383413128b2e6d5ee2cefe920a4f89 100644
--- a/tests/test_public_controller.py
+++ b/tests/test_public_controller.py
@@ -20,7 +20,7 @@
 from __future__ import absolute_import
 
 # from aura_engine_api.rest.models.track import Track  # noqa: E501
-from aura_engine_api.rest.test import BaseTestCase
+from tests import BaseTestCase
 
 # from flask import json
 # from six import BytesIO