diff --git a/.github/workflows/vite-build.yml b/.github/workflows/vite-build.yml
index f3aa726..71f3d49 100644
--- a/.github/workflows/vite-build.yml
+++ b/.github/workflows/vite-build.yml
@@ -3,7 +3,7 @@ name: Build frontend and publish to Comfy registry
 on:
   push:
     branches:
-      - release/comfy_node
+      - main
   pull_request:
     branches:
       - main
@@ -42,7 +42,7 @@ jobs:
         run: pnpm run build
 
       - name: Publish Custom Node
-        if: ${{ github.event_name == 'push' && github.ref == 'refs/heads/release/comfy_node' }}
+        if: ${{ github.event_name == 'push' && github.ref == 'refs/heads/main' }}
         uses: Comfy-Org/publish-node-action@main
         with:
           personal_access_token: ${{ secrets.REGISTRY_ACCESS_TOKEN }}