diff --git a/.github/container/Dockerfile.maxtext.amd64 b/.github/container/Dockerfile.maxtext.amd64
index 8289a6099..96f06f022 100644
--- a/.github/container/Dockerfile.maxtext.amd64
+++ b/.github/container/Dockerfile.maxtext.amd64
@@ -21,7 +21,8 @@ EOF
 ## Add test script to the path
 ###############################################################################
 
-ADD test-maxtext.sh /usr/local/bin
+ADD .github/container/test-maxtext.sh /usr/local/bin
+ADD rosetta /opt/rosetta
 
 ###############################################################################
 ## Install accumulated packages from the base image and the previous stage
@@ -31,4 +32,4 @@ FROM mealkit as final
 
 RUN pip-finalize.sh
 
-WORKDIR ${SRC_PATH_MAXTEXT}
\ No newline at end of file
+WORKDIR ${SRC_PATH_MAXTEXT}
diff --git a/.github/container/Dockerfile.maxtext.arm64 b/.github/container/Dockerfile.maxtext.arm64
index bd64c5b6d..2ff52cc4a 100644
--- a/.github/container/Dockerfile.maxtext.arm64
+++ b/.github/container/Dockerfile.maxtext.arm64
@@ -62,7 +62,8 @@ EOF
 ## Add test script to the path
 ###############################################################################
 
-ADD test-maxtext.sh /usr/local/bin
+ADD .github/container/test-maxtext.sh /usr/local/bin
+ADD rosetta /opt/rosetta
 
 ###############################################################################
 ## Install accumulated packages from the base image and the previous stage
diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
index 426764323..91089d9d6 100644
--- a/.github/workflows/_ci.yaml
+++ b/.github/workflows/_ci.yaml
@@ -108,6 +108,7 @@ jobs:
       BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAG_MEALKIT }}
       CONTAINER_NAME: maxtext
       DOCKERFILE: .github/container/Dockerfile.maxtext.${{ inputs.ARCHITECTURE }}
+      DOCKER_CONTEXT: .
       EXTRA_BUILD_ARGS: |
         URLREF_MAXTEXT=${{ fromJson(inputs.SOURCE_URLREFS).MAXTEXT }}
     secrets: inherit