diff --git a/.github/workflows/nightly-rosetta-pax-build.yaml b/.github/workflows/nightly-rosetta-pax-build.yaml
index a2ff8ffc7..9a344b7d7 100644
--- a/.github/workflows/nightly-rosetta-pax-build.yaml
+++ b/.github/workflows/nightly-rosetta-pax-build.yaml
@@ -99,6 +99,7 @@ jobs:
     needs: [metadata, amd64, arm64]
     uses: ./.github/workflows/_publish_badge.yaml
     if: always()
+    with:
       ENDPOINT_FILENAME: 'rosetta-pax-build-status.json'
       PUBLISH: ${{ needs.metadata.outputs.PUBLISH == 'true' }}
       SCRIPT: |
@@ -152,7 +153,7 @@ jobs:
         type=raw,value=nightly-${{ needs.metadata.outputs.BUILD_DATE }},priority=900
 
   publish-test-badge:
-    needs: [metadata, publish-build-badge, test-amd64]
+    needs: [metadata, amd64, arm64, test-amd64]
     uses: ./.github/workflows/_publish_badge.yaml
     if: always()
     secrets: inherit
diff --git a/.github/workflows/nightly-rosetta-t5x-build-test.yaml b/.github/workflows/nightly-rosetta-t5x-build-test.yaml
index 96088e728..570069992 100644
--- a/.github/workflows/nightly-rosetta-t5x-build-test.yaml
+++ b/.github/workflows/nightly-rosetta-t5x-build-test.yaml
@@ -178,7 +178,7 @@ jobs:
     secrets: inherit
 
   publish-test-badge:
-    needs: [metadata, publish-build-badge, test-unit-amd64, test-t5x-amd64, test-vit-amd64]
+    needs: [metadata, amd64, arm64, test-unit-amd64, test-t5x-amd64, test-vit-amd64]
     uses: ./.github/workflows/_publish_badge.yaml
     if: always()
     secrets: inherit