diff --git a/resolvers/notifications.py b/resolvers/notifications.py
index e11277c6..dcc3945b 100644
--- a/resolvers/notifications.py
+++ b/resolvers/notifications.py
@@ -1,9 +1,9 @@
 from sqlalchemy import select, desc, and_, update
 
 from auth.credentials import AuthCredentials
-from base.resolvers import query, mutation
+from services.schema import query, mutation
 from auth.authenticate import login_required
-from base.orm import local_session
+from services.db import local_session
 from orm import Notification
 
 
diff --git a/services/notifications/notification_service.py b/services/notifications/notification_service.py
index bfaf8e79..294eadb4 100644
--- a/services/notifications/notification_service.py
+++ b/services/notifications/notification_service.py
@@ -4,7 +4,7 @@ from datetime import datetime, timezone
 
 from sqlalchemy import and_
 
-from base.orm import local_session
+from services.db import local_session
 from orm import Reaction, Shout, Notification, User
 from orm.notification import NotificationType
 from orm.reaction import ReactionKind