diff --git a/trex_core/application/snapshot_scheduler.py b/trex_core/application/snapshot_scheduler.py
index e1d23f2..b6ef0bc 100644
--- a/trex_core/application/snapshot_scheduler.py
+++ b/trex_core/application/snapshot_scheduler.py
@@ -22,6 +22,7 @@ from trex_core.application.portfolio import (
     DegradedReason,
     PortfolioHubReadPort,
     PortfolioHubUnavailableError,
+    ReconcileNotificationStore,
     ReconcileWarningStore,
     SnapshotStore,
     TradingContext,
@@ -49,7 +50,7 @@ class SnapshotScheduler:
         instance_key: str = "default",
         alarm_emitter: Optional[AlarmEmitter] = None,
         warn_store: Optional[ReconcileWarningStore] = None,
-        notification_store=None,
+        notification_store: Optional[ReconcileNotificationStore] = None,
         credential_fetch_enabled: bool = False,
         kis_reinit_callback: Optional[Callable[[BrokerCredentialPayload], None]] = None,
         credential_refresh: Optional[Callable[[], object]] = None,