diff --git a/services/keeper.py b/services/keeper.py
index 8721d2c..3b658b1 100644
--- a/services/keeper.py
+++ b/services/keeper.py
@@ -8,7 +8,7 @@ from services.db import local_session
 from orm.shout import Shout
 from orm.author import Author
 from orm.user import User
-from orm.notification import NotificationType, Notification
+from orm.notification import NotificationAction, NotificationEntity, Notification
 from orm.reaction import ReactionKind, Reaction