diff --git a/tests/test_databases_sqlite.py b/tests/test_databases_sqlite.py
index 9def093..7957737 100644
--- a/tests/test_databases_sqlite.py
+++ b/tests/test_databases_sqlite.py
@@ -6,7 +6,8 @@
 
 from neon_users_service.databases.sqlite import SQLiteUserDatabase
 from neon_users_service.exceptions import UserExistsError, UserNotFoundError
-from neon_users_service.models import User, AccessRoles
+from neon_data_models.models.user import User
+from neon_data_models.enum import AccessRoles
 
 
 class TestSqlite(TestCase):
diff --git a/tests/test_service.py b/tests/test_service.py
index f53b867..1d98d65 100644
--- a/tests/test_service.py
+++ b/tests/test_service.py
@@ -7,7 +7,7 @@
 from neon_users_service.databases.sqlite import SQLiteUserDatabase
 from neon_users_service.exceptions import ConfigurationError, AuthenticationError, UserNotFoundError, \
     UserNotMatchedError
-from neon_users_service.models import User
+from neon_data_models.models.user import User
 from neon_users_service.service import NeonUsersService