diff --git a/comfy_api_nodes/apis/client.py b/comfy_api_nodes/apis/client.py
index cc135ad0a..92aab9435 100644
--- a/comfy_api_nodes/apis/client.py
+++ b/comfy_api_nodes/apis/client.py
@@ -108,7 +108,8 @@ from comfy.cli_args import args
 from comfy import utils
 from . import request_logger
 
-# Import models from your generated stubs
+from comfy.cli_args import args
+from comfy import utils
 
 T = TypeVar("T", bound=BaseModel)
 R = TypeVar("R", bound=BaseModel)