diff --git a/docker/run/Dockerfile b/docker/run/Dockerfile
index f3f813e0f..f56d4f901 100644
--- a/docker/run/Dockerfile
+++ b/docker/run/Dockerfile
@@ -1,7 +1,7 @@
 # Use the pre-built base image for A0
 # FROM agent-zero-base:local
-FROM agent0ai/agent-zero-base:testing
-# FROM agent0ai/agent-zero-base:latest
+# FROM agent0ai/agent-zero-base:testing
+FROM agent0ai/agent-zero-base:latest
 
 # Check if the argument is provided, else throw an error
 ARG BRANCH