diff --git a/python/aibrix/aibrix/gpuoptimizer/loadmonitor/monitor.py b/python/aibrix/aibrix/gpuoptimizer/loadmonitor/monitor.py
index deb6e873..1a566924 100644
--- a/python/aibrix/aibrix/gpuoptimizer/loadmonitor/monitor.py
+++ b/python/aibrix/aibrix/gpuoptimizer/loadmonitor/monitor.py
@@ -88,7 +88,7 @@ def __init__(
             debug: (optional) Whether to enable debugging behavior. Defaults to False.
         """
         self.model_name = model_name
-        self.deployments: dict[str, DeploymentStates] = {}
+        self.deployments: Dict[str, DeploymentStates] = {}
         self.thread = None
         self.outdated_watch_version = None
         self.last_watch_version = None