diff --git a/.github/workflows/docs-amplify.yaml b/.github/workflows/docs-amplify.yaml
index 15eff5d8db906..f50fab062cfbb 100644
--- a/.github/workflows/docs-amplify.yaml
+++ b/.github/workflows/docs-amplify.yaml
@@ -12,7 +12,7 @@ permissions:
 jobs:
   amplify-preview:
     name: Get and post Amplify preview URL
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04-2core-arm64
     environment: docs-amplify
     steps:
     - name: Checkout shared-workflow
@@ -29,8 +29,10 @@ jobs:
         aws-region: us-west-2
         role-to-assume: ${{ vars.IAM_ROLE }}
 
-    - uses: ./tools/amplify-preview
+    - name: Amplify Preview
+      uses: ./tools/amplify-preview
       with:
         app_ids: ${{ vars.AMPLIFY_APP_IDS }}
         create_branches: "true"
         github_token: ${{ secrets.GITHUB_TOKEN }}
+        wait: "true"