diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
index c392cdd9..1a7d1eab 100644
--- a/.github/workflows/pr.yaml
+++ b/.github/workflows/pr.yaml
@@ -15,7 +15,8 @@ jobs:
     needs:
       - checks
       - wheel-build
-      - wheel-test
+      - wheel-test-amd64
+      - wheel-test-arm64
     secrets: inherit
     uses: rapidsai/shared-action-workflows/.github/workflows/pr-builder.yaml@branch-23.12
   checks: