diff --git a/templates/compose/affine.yaml b/templates/compose/affine.yaml
index 9d31d3be0..92231d41a 100644
--- a/templates/compose/affine.yaml
+++ b/templates/compose/affine.yaml
@@ -7,7 +7,7 @@
 
 services:
   affine:
-    image: ghcr.io/toeverything/affine-graphql:stable
+    image: ghcr.io/toeverything/affine:stable
     depends_on:
       redis:
         condition: service_healthy
@@ -44,7 +44,7 @@ services:
 
   affine_migration:
     exclude_from_hc: true
-    image: ghcr.io/toeverything/affine-graphql:${AFFINE_REVISION:-stable}
+    image: ghcr.io/toeverything/affine:${AFFINE_REVISION:-stable}
     restart: on-failure
     volumes:
       - affine-storage:/root/.affine/storage
@@ -75,7 +75,7 @@ services:
       timeout: 5s
       retries: 5
   postgres:
-    image: postgres:16
+    image: pgvector/pgvector:pg16
     volumes:
       - affine-postgres-data:/var/lib/postgresql/data
     healthcheck: