diff --git a/.github/workflows/auto-merge.yml b/.github/workflows/auto-merge.yml
index 4f80aa5..ab76ce6 100644
--- a/.github/workflows/auto-merge.yml
+++ b/.github/workflows/auto-merge.yml
@@ -110,7 +110,7 @@ jobs:
           fi
 
       - name: prepare code base if $FILE_USE_BASE has content
-        uses: NVIDIA/spark-rapids-common/checkout@main
+        uses: YanxuanLiu/spark-rapids-common/checkout@checkout-wrapper
         if: ${{ inputs.file_use_base != '' && env.continue_merge == 'true' }}
         with:
           ref: ${{ env.HEAD }} # force to fetch from the latest upstream instead of PR ref
diff --git a/.github/workflows/license-header-check.yml b/.github/workflows/license-header-check.yml
index 102588f..1c48a92 100644
--- a/.github/workflows/license-header-check.yml
+++ b/.github/workflows/license-header-check.yml
@@ -29,7 +29,7 @@ jobs:
           echo "PR_FETCH_DEPTH=$(( ${{ github.event.pull_request.commits }} + 10 ))" >> $GITHUB_ENV
 
       - name: Checkout code
-        uses: NVIDIA/spark-rapids-common/checkout@main
+        uses: YanxuanLiu/spark-rapids-common/checkout@checkout-wrapper 
         with:
           fetch-depth: ${{ env.PR_FETCH_DEPTH }}
 
diff --git a/.github/workflows/shell-check.yml b/.github/workflows/shell-check.yml
index 860c724..c200904 100644
--- a/.github/workflows/shell-check.yml
+++ b/.github/workflows/shell-check.yml
@@ -29,7 +29,7 @@ jobs:
           echo "PR_FETCH_DEPTH=$(( ${{ github.event.pull_request.commits }} + 10 ))" >> $GITHUB_ENV
 
       - name: Checkout code
-        uses: NVIDIA/spark-rapids-common/checkout@main
+        uses: YanxuanLiu/spark-rapids-common/checkout@checkout-wrapper
         with:
           fetch-depth: ${{ env.PR_FETCH_DEPTH }}
 
diff --git a/signoff-check/action.yml b/signoff-check/action.yml
index c4635f5..2c636b8 100644
--- a/signoff-check/action.yml
+++ b/signoff-check/action.yml
@@ -36,7 +36,7 @@ runs:
   using: "composite"
   steps:
     - name: Checkout repository
-      uses: NVIDIA/spark-rapids-common/checkout@main
+      uses: YanxuanLiu/spark-rapids-common/checkout@checkout-wrapper
       with:
         repository: NVIDIA/spark-rapids-common
         ref: main