diff --git a/templates/compose/rivet-engine.yaml b/templates/compose/rivet-engine.yaml
index 608cdaac3..e77561cca 100644
--- a/templates/compose/rivet-engine.yaml
+++ b/templates/compose/rivet-engine.yaml
@@ -7,14 +7,13 @@
 
 services:
   rivet-engine:
-    image: rivetkit/engine:25.8.0
+    image: rivetdev/engine:2.2.0
     environment:
       - SERVICE_URL_RIVET_6420
-      - 'RIVET__AUTH__ADMIN_TOKEN=${SERVICE_PASSWORD_RIVET}'
-      - RIVET__POSTGRES__URL=postgresql://$SERVICE_USER_POSTGRESQL:$SERVICE_PASSWORD_POSTGRESQL@postgresql:5432/${POSTGRESQL_DATABASE-rivet}
-    depends_on:
-      postgresql:
-        condition: service_healthy
+      - RIVET__FILE_SYSTEM__PATH=/data
+      - 'RIVET__AUTH__ADMIN_TOKEN=${SERVICE_BASE64_TOKEN}'
+    volumes:
+      - 'rivet-data:/data'
     healthcheck:
       test:
         - CMD
@@ -24,19 +23,4 @@ services:
       interval: 2s
       timeout: 10s
       retries: 10
-      start_period: 30s
-
-  postgresql:
-    image: postgres:17-alpine
-    volumes:
-      - rivet-postgresql-data:/var/lib/postgresql/data
-    environment:
-      - POSTGRES_USER=${SERVICE_USER_POSTGRESQL}
-      - POSTGRES_PASSWORD=${SERVICE_PASSWORD_POSTGRESQL}
-      - POSTGRES_DB=${POSTGRESQL_DATABASE-rivet}
-    healthcheck:
-      test: ["CMD-SHELL", "pg_isready -U $${POSTGRES_USER} -d $${POSTGRES_DB}"]
-      interval: 5s
-      timeout: 20s
-      retries: 10
-
+      start_period: 30s
\ No newline at end of file