diff --git a/compose.yml b/compose.yml
index 6533b94da..8a4bfcaae 100644
--- a/compose.yml
+++ b/compose.yml
@@ -2,6 +2,7 @@
 x-comfyui-common: &comfyui-common
   build:
     context: .
+  image: comfyui
   ports:
     - "${PORT:-8188}:8188"
   volumes:
@@ -11,13 +12,13 @@ x-comfyui-common: &comfyui-common
 
 # ComfyUI profiles
 services:
-  cpu:
+  comfyui-cpu:
     <<: *comfyui-common
     command: >
       python main.py --listen=${LISTEN_IP:-0.0.0.0} --cpu
     profiles:
       - cpu
-  cuda:
+  comfyui-cuda:
     <<: *comfyui-common
     command: >
       python main.py --listen=${LISTEN_IP:-0.0.0.0}