diff --git a/rosetta/Dockerfile.maxtext b/rosetta/Dockerfile.maxtext
index 936fbefa1..54377dccf 100644
--- a/rosetta/Dockerfile.maxtext
+++ b/rosetta/Dockerfile.maxtext
@@ -1,5 +1,5 @@
 # syntax=docker/dockerfile:1-labs
-ARG BASE_IMAGE=ghcr.io/nvidia/jax-mealkit:upstream-pax
+ARG BASE_IMAGE=ghcr.io/nvidia/jax-mealkit:upstream-maxtext
 ARG GIT_USER_EMAIL=jax@nvidia.com
 ARG GIT_USER_NAME=NVIDIA
 # If set to "true", then will pull new local patches, the manifest.yaml and create-distribution.sh (in case it was updated).