diff --git a/netx_api/ume_alarm_ws_subscription.py b/netx_api/ume_alarm_ws_subscription.py
index 4e2a77c..e84ffa4 100644
--- a/netx_api/ume_alarm_ws_subscription.py
+++ b/netx_api/ume_alarm_ws_subscription.py
@@ -10,6 +10,7 @@ from sqlalchemy.orm import Session
 
 from . import ume_alarm_ws_state as ws_state
 from .config import settings
+from .db import SessionLocal
 from .ume_alarm_subscription_store import (
     DEFAULT_SUBSCRIPTION_KEY,
     clear_subscription,