diff --git a/docker-compose.cpu.yml b/docker-compose.cpu.yml
index a78417c..707d58f 100644
--- a/docker-compose.cpu.yml
+++ b/docker-compose.cpu.yml
@@ -26,7 +26,7 @@ services:
       start_period: 1s
 
   kokoro-tts:
-    image: ghcr.io/remsky/kokoro-fastapi:latest-cpu
+    image: ghcr.io/remsky/kokoro-fastapi-cpu:latest
     # Uncomment below (and comment out above) to build from source instead of using the released image
     # build:
       # context: .
@@ -52,7 +52,7 @@ services:
 
   # Gradio UI service [Comment out everything below if you don't need it]
   gradio-ui:
-    image: ghcr.io/remsky/kokoro-fastapi:latest-ui
+    image: ghcr.io/remsky/kokoro-fastapi-ui:latest
     # Uncomment below (and comment out above) to build from source instead of using the released image
     build:
       context: ./ui
diff --git a/docker-compose.yml b/docker-compose.yml
index 286cd68..3e4bd3d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -32,7 +32,7 @@ services:
       start_period: 1s
 
   kokoro-tts:
-    image: ghcr.io/remsky/kokoro-fastapi:latest
+    image: ghcr.io/remsky/kokoro-fastapi-gpu:latest
     # Uncomment below (and comment out above) to build from source instead of using the released image
     # build:
       # context: .
@@ -62,7 +62,7 @@ services:
 
   # Gradio UI service [Comment out everything below if you don't need it]
   gradio-ui:
-    image: ghcr.io/remsky/kokoro-fastapi:latest-ui
+    image: ghcr.io/remsky/kokoro-fastapi-ui:latest
     # Uncomment below (and comment out above) to build from source instead of using the released image
     # build:
       # context: ./ui