diff --git a/.github/workflows/gradle.yml b/.github/workflows/gradle.yml
index ded43f5..4312837 100644
--- a/.github/workflows/gradle.yml
+++ b/.github/workflows/gradle.yml
@@ -2,9 +2,9 @@ name: Java CI
 
 on:
   push:
-    branches: [ $default-branch ]
+    branches: [ main ]
   pull_request:
-    branches: [ $default-branch ]
+    branches: [ main ]
 
 jobs:
   build:
@@ -16,9 +16,10 @@ jobs:
       - name: Set up JDK 16
         uses: actions/setup-java@v2
         with:
-          java-version: '16'
+          java-version: 16
           distribution: 'temurin'
+          cache: gradle
       - name: Build with Gradle
-        uses: gradle/gradle-build-action@4137be6a8bf7d7133955359dbd952c0ca73b1021
+        uses: gradle/gradle-build-action@v2
         with:
           arguments: build