diff --git a/backend/alembic/env.py b/backend/alembic/env.py
index 5c89ae8e5..fc6d3225a 100644
--- a/backend/alembic/env.py
+++ b/backend/alembic/env.py
@@ -31,6 +31,7 @@
 from app.models.tool import Tool  # noqa: F401
 from app.models.trigger import AgentTrigger  # noqa: F401
 from app.models.agent_credential import AgentCredential  # noqa: F401
+from app.models.notification import Notification  # noqa: F401
 
 config = context.config
 settings = get_settings()