diff --git a/.github/workflows/test-ui.yaml b/.github/workflows/test-ui.yaml
index 68758670d..ad50fcd50 100644
--- a/.github/workflows/test-ui.yaml
+++ b/.github/workflows/test-ui.yaml
@@ -21,6 +21,7 @@ jobs:
       with:
         repository: "huchenlei/ComfyUI_frontend"
         path: "ComfyUI_frontend"
+        ref: ${{ github.head_ref }}
     - name: Get commit message
       id: commit-message
       run: echo "::set-output name=message::$(git log -1 --pretty=%B)"