diff --git a/rnd/docker-compose.yml b/rnd/docker-compose.yml
index 0e628a0e95..4a37096994 100644
--- a/rnd/docker-compose.yml
+++ b/rnd/docker-compose.yml
@@ -16,7 +16,10 @@ services:
       - app-network
 
   migrate:
-    image: autogpt_server:latest
+    build:
+      context: ../
+      dockerfile: rnd/autogpt_server/Dockerfile
+      target: server
     command: ["sh", "-c", "poetry run prisma migrate deploy"]
     develop:
       watch: