diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index ea16aff..b3e5caa 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -30,7 +30,7 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Setup uv
-        uses: astral-sh/setup-uv@v6
+        uses: ultralytics/actions/setup-uv@main
 
       - name: Install dependencies
         run: |
diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
index 672e295..49678bb 100644
--- a/.github/workflows/push.yml
+++ b/.github/workflows/push.yml
@@ -31,7 +31,7 @@ jobs:
         with:
           python-version: 3.12
       - name: Setup uv
-        uses: astral-sh/setup-uv@v6
+        uses: ultralytics/actions/setup-uv@main
       - name: Install dependencies
         uses: ultralytics/actions/retry@main
         with: