diff --git a/tests/uitests/build.gradle.kts b/tests/uitests/build.gradle.kts
index 53876b0c8c..466167981c 100644
--- a/tests/uitests/build.gradle.kts
+++ b/tests/uitests/build.gradle.kts
@@ -20,7 +20,7 @@ android {
     namespace = "ui"
 }
 
-tasks.withType(Test::class.java) {
+tasks.withType(Test::class) {
     // Don't fail the test run if there are no tests, this can happen if we run them with screenshot test disabled
     failOnNoDiscoveredTests = false
 }