diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 50f2b4d..5fe62ca 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -10,7 +10,7 @@ on:
 jobs:
   build-archive-wheel:
     
-    uses: OpenBMB/BMTrain/.github/workflows/build_whl.yml@workflow
+    uses: OpenBMB/BMTrain/.github/workflows/build_whl.yml@main
     secrets:
       DOCKERHUB_TOKEN: ${{ secrets.DOCKERHUB_TOKEN }}
       DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_USERNAME }}
diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index 98e579f..043afe2 100644
--- a/.github/workflows/publish.yaml
+++ b/.github/workflows/publish.yaml
@@ -9,7 +9,7 @@ on:
 jobs:
 
   build-archive-wheel:
-    uses: OpenBMB/BMTrain/.github/workflows/build_whl.yml@workflow
+    uses: OpenBMB/BMTrain/.github/workflows/build_whl.yml@main
     secrets:
       DOCKERHUB_TOKEN: ${{ secrets.DOCKERHUB_TOKEN }}
       DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_USERNAME }}
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index c5fd193..d6c8f62 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -9,7 +9,7 @@ jobs:
 
   build-archive-wheel:
 
-    uses: OpenBMB/BMTrain/.github/workflows/build_whl.yml@workflow
+    uses: OpenBMB/BMTrain/.github/workflows/build_whl.yml@main
     secrets:
       DOCKERHUB_TOKEN: ${{ secrets.DOCKERHUB_TOKEN }}
       DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_USERNAME }}