diff --git a/.github/workflows/_sandbox.yaml b/.github/workflows/_sandbox.yaml
index 6bb139dfa..65f65329e 100644
--- a/.github/workflows/_sandbox.yaml
+++ b/.github/workflows/_sandbox.yaml
@@ -62,6 +62,6 @@ jobs:
     with:
       ARCHITECTURE: amd64
       BUILD_DATE: "2023-11-15"
-      BASE_IMAGE: ghcr.io/nvidia/jax-toolbox-internal:6896408819-upstream-pax-amd64-mealkit
+      BASE_IMAGE: ghcr.io/nvidia/jax-toolbox-internal:6896408819-upstream-pax-amd64
       BASE_LIBRARY: pax
     secrets: inherit
diff --git a/rosetta/Dockerfile.pax b/rosetta/Dockerfile.pax
index 503be8dcb..39f719769 100644
--- a/rosetta/Dockerfile.pax
+++ b/rosetta/Dockerfile.pax
@@ -1,9 +1,14 @@
 # syntax=docker/dockerfile:1-labs
-ARG BASE_IMAGE=ghcr.io/nvidia/upstream-pax:mealkit
+ARG BASE_IMAGE=ghcr.io/nvidia/upstream-pax:latest
 ARG GIT_USER_EMAIL=jax@nvidia.com
 ARG GIT_USER_NAME=NVIDIA
+ARG SRC_PATH_PAXML=/opt/paxml
+ARG SRC_PATH_PRAXIS=/opt/praxis
 
 FROM scratch as rosetta-source
+ARG SRC_PATH_PAXML
+ARG SRC_PATH_PRAXIS
+
 COPY . /
 
 FROM scratch as pax-mirror-source
@@ -25,7 +30,7 @@ EOF
 COPY --from=rosetta-source / /opt/rosetta
 WORKDIR /opt/rosetta
 RUN --mount=target=/opt/pax-mirror,from=pax-mirror-source,readwrite \
-    --mount=target=/opt/praxis-mirror,from=praxis-mirror-source,readwrite <<"EOF" bash -e
+    --mount=target=/opt/praxis-mirror,from=praxis-mirror-source,readwrite <<EOF bash -e
 bash create-distribution.sh \
   -p patchlist-paxml.txt \
   -m https://github.com/nvjax-svc-0/paxml.git \