diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index f4df3fd..7276531 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -30,16 +30,18 @@ jobs:
       matrix:
         os: [ubuntu-latest, windows-latest, macos-14]
     steps:
-      - name: Set up repository
-        uses: tree-sitter/parser-setup-action@v1.2
-        with:
-          node-version: ${{vars.NODE_VERSION}}
+      - name: Checkout repository
+        uses: actions/checkout@v4
+      - name: Install tree-sitter CLI
+        uses: tree-sitter/setup-action/cli@v1
       - name: Build with all extensions
         run: npm run build
         env:
           ALL_EXTENSIONS: 1
       - name: Run tests
-        uses: tree-sitter/parser-test-action@v1.2
+        uses: tree-sitter/parser-test-action@v2
+        with:
+          test-rust: true
       - name: Rebuild with default extensions
         run: npm run build
       - name: Verify grammar consistency