diff --git a/.github/actions/setup-build/action.yml b/.github/actions/setup-build/action.yml
index 2fa907a01..80adf5281 100644
--- a/.github/actions/setup-build/action.yml
+++ b/.github/actions/setup-build/action.yml
@@ -328,15 +328,17 @@ runs:
 
     - name: Setup Swift toolchain (Upstream)
       if: inputs.swift-version != '' && inputs.swift-repo == ''
-      uses: compnerd/gha-setup-swift@6c9f2db7c3155c57fe35f160bcd5cf5859b9c1ba # main
+      uses: compnerd/gha-setup-swift@main
       with:
-        branch: ${{ steps.sanitize-input.outputs.swift-branch }}
-        tag: ${{ steps.sanitize-input.outputs.swift-tag }}
+        source: 'swift.org'
+        swift-version: ${{ steps.sanitize-input.outputs.swift-branch }}
+        swift-build: ${{ steps.sanitize-input.outputs.swift-tag }}
 
     - name: Setup Swift toolchain (Custom)
       if: inputs.swift-version != '' && inputs.swift-repo != ''
-      uses: compnerd/gha-setup-swift@6c9f2db7c3155c57fe35f160bcd5cf5859b9c1ba # main
+      uses: compnerd/gha-setup-swift@main
       with:
+        source: 'custom'
         github-repo: ${{ inputs.swift-repo }}
         github-token: ${{ github.token }}
         release-asset-name: ${{ steps.sanitize-input.outputs.swift-release-asset }}