diff --git a/ollama/compose.yml b/ollama/compose.yml
index f0bef95..d9a7bee 100644
--- a/ollama/compose.yml
+++ b/ollama/compose.yml
@@ -16,7 +16,8 @@ services:
       - ollama
     ports:
       - ${OPEN_WEBUI_PORT-3001}:8080
-
+    extra_hosts:
+      - host.docker.internal:host-gateway   
     volumes:
       - open-webui:/app/backend/data
     container_name: open-webui
@@ -31,7 +32,7 @@ services:
       # comfy
       - ENABLE_IMAGE_GENERATION=True
       - IMAGE_GENERATION_ENGINE=comfyui
-      - "COMFYUI_BASE_URL=http://127.0.0.1:8188"
+      - "COMFYUI_BASE_URL=http://host.docker.internal:8188"
 
 volumes:
   open-webui: