diff --git a/comfy/model_management.py b/comfy/model_management.py
index e89f80d6..3aea7ea8 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -1,6 +1,6 @@
 import psutil
 from enum import Enum
-from .cli_args import args
+from comfy.cli_args import args
 
 class VRAMState(Enum):
     CPU = 0