diff --git a/samples/hello-spring-boot/build.gradle.kts b/samples/hello-spring-boot/build.gradle.kts
index 5089cf894..7e3db2cbf 100644
--- a/samples/hello-spring-boot/build.gradle.kts
+++ b/samples/hello-spring-boot/build.gradle.kts
@@ -18,7 +18,9 @@ tasks.withType<KotlinCompile> {
     }
 }
 
-(tasks.findByName("test") as Test).useJUnitPlatform()
+tasks.withType<Test> {
+    useJUnitPlatform()
+}
 
 repositories {
     mavenCentral()