diff --git a/.github/workflows/bump_torch_mlir.yml b/.github/workflows/bump_torch_mlir.yml
index 594c32761826c..85fb1d42e30e1 100644
--- a/.github/workflows/bump_torch_mlir.yml
+++ b/.github/workflows/bump_torch_mlir.yml
@@ -49,8 +49,8 @@ jobs:
         run: |
           cd third_party/torch-mlir
           git fetch origin
-          if [ "${{ github.event_name }}" == "workflow_dispatch" ]; then
-            git checkout ${{ github.event.inputs.bump_target }}
+          if [ "workflow_dispatch" == "workflow_dispatch" ]; then
+            git checkout upstream/main || { echo "Checkout failed. Available remotes:"; git remote -v; exit 1; }
           else
             git checkout origin/main
           fi