diff --git a/comfy/cmd/server.py b/comfy/cmd/server.py
index 2de115142..9853d55ac 100644
--- a/comfy/cmd/server.py
+++ b/comfy/cmd/server.py
@@ -2,6 +2,8 @@ from __future__ import annotations
 import asyncio
 import glob
 import struct
+import sys
+
 from PIL import Image, ImageOps
 from io import BytesIO