diff --git a/workers/comfyui-json/server.py b/workers/comfyui-json/server.py
index a2c0438..45b3570 100644
--- a/workers/comfyui-json/server.py
+++ b/workers/comfyui-json/server.py
@@ -12,7 +12,7 @@ from lib.server import start_server
 from .data_types import ComfyWorkflowData
 
 
-MODEL_SERVER_URL = "http://127.0.0.1:18288"
+MODEL_SERVER_URL = os.getenv("MODEL_SERVER_URL", "http://127.0.0.1:18288")
 
 # This is the last log line that gets emitted once comfyui+extensions have been fully loaded
 MODEL_SERVER_START_LOG_MSG = "To see the GUI go to: http://127.0.0.1:18188"