diff --git a/comfyui_manager/__init__.py b/comfyui_manager/__init__.py
index 4c814674..386bd2eb 100644
--- a/comfyui_manager/__init__.py
+++ b/comfyui_manager/__init__.py
@@ -1,5 +1,6 @@
 import os
 import logging
+from comfy.cli_args import args
 
 ENABLE_LEGACY_COMFYUI_MANAGER_FRONT_DEFAULT = True # Enable legacy ComfyUI Manager frontend while new UI is in beta phase