diff --git a/.github/workflows/test-ui.yaml b/.github/workflows/test-ui.yaml
index 514f090fc..f31dd7e97 100644
--- a/.github/workflows/test-ui.yaml
+++ b/.github/workflows/test-ui.yaml
@@ -9,9 +9,9 @@ jobs:
     - name: Checkout ComfyUI
       uses: actions/checkout@v4
       with:
-        repository: "huchenlei/ComfyUI"
+        repository: "comfyanonymous/ComfyUI"
         path: "ComfyUI"
-        ref: main
+        ref: master
     - name: Checkout ComfyUI_frontend
       uses: actions/checkout@v4
       with: