diff --git a/.github/workflows/java.yaml b/.github/workflows/java.yaml
index 077fb3e..c454b0a 100644
--- a/.github/workflows/java.yaml
+++ b/.github/workflows/java.yaml
@@ -9,14 +9,14 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - name: Set up JDK 21
-        uses: actions/setup-java@v4
+        uses: actions/setup-java@v5
         with:
           distribution: "temurin"
           java-version: "21"
 
       - uses: gradle/actions/setup-gradle@v4
         with:
-          gradle-version: 8.14.1
+          gradle-version: 9.1.0
       - name: Run test
         run: |
           gradle generateAvro
diff --git a/build.gradle b/build.gradle
index e566c18..043841b 100644
--- a/build.gradle
+++ b/build.gradle
@@ -4,6 +4,7 @@ plugins {
     id "com.github.davidmc24.gradle.plugin.avro" version "1.9.1"
 }
 
+wrapper { gradleVersion = '9.1.0' }
 apply plugin: "java"
 apply plugin: 'maven-publish'
 apply plugin: "com.github.davidmc24.gradle.plugin.avro"
@@ -36,15 +37,24 @@ dependencies {
     testImplementation group: 'org.junit.jupiter', name: 'junit-jupiter-params', version: '5.11.3'
     testImplementation group: 'org.junit.jupiter', name: 'junit-jupiter-engine', version: '5.11.3'
     testImplementation group: 'org.assertj', name: 'assertj-core', version: '3.26.3'
-    test.useJUnitPlatform()
+
+    testRuntimeOnly "org.junit.platform:junit-platform-launcher"
 
 }
 
 tasks.withType(JavaCompile) {
     options.encoding = 'UTF-8'
+
 }
 
 
+
+test{
+    useJUnit()
+    useJUnitPlatform()
+
+}
+
 task generateAvro(type: com.github.davidmc24.gradle.plugin.avro.GenerateAvroJavaTask) {
     stringType = "CharSequence"
     source("src/test/resources/avro")
@@ -61,9 +71,4 @@ sourceSets {
             srcDir 'build/generated-main-avro-java/'
         }
     }
-    test {
-        java {
-            srcDir 'src/test/java'
-        }
-    }
 }
\ No newline at end of file