diff --git a/idea/idea-gradle/tests/org/jetbrains/kotlin/gradle/GradleMultiplatformHighlightingTest.kt b/idea/idea-gradle/tests/org/jetbrains/kotlin/gradle/GradleMultiplatformHighlightingTest.kt
index 3589ed3190c..bd4d238f5ec 100644
--- a/idea/idea-gradle/tests/org/jetbrains/kotlin/gradle/GradleMultiplatformHighlightingTest.kt
+++ b/idea/idea-gradle/tests/org/jetbrains/kotlin/gradle/GradleMultiplatformHighlightingTest.kt
@@ -41,12 +41,10 @@ class GradleMultiplatformHighlightingTest : GradleImportingTestCase() {
         val project = myTestFixture.project
 
         checkFiles(
-            files,
+            files.filter { it.extension == "kt" },
             project,
             object : GradleDaemonAnalyzerTestCase(testLineMarkers = true, checkWarnings = true, checkInfos = false) {}
-        ) { file ->
-            file.extension == "kt"
-        }
+        )
     }
 
     override fun testDataDirName(): String {
@@ -72,19 +70,17 @@ abstract class GradleDaemonAnalyzerTestCase(
 internal fun checkFiles(
     files: List<VirtualFile>,
     project: Project,
-    analyzer: GradleDaemonAnalyzerTestCase,
-    fileFilter: (VirtualFile) -> Boolean
+    analyzer: GradleDaemonAnalyzerTestCase
 ) {
     var atLeastOneFile = false
-    val kotlinFiles = files.filter(fileFilter)
     val content = mutableMapOf<VirtualFile, String>()
-    kotlinFiles.forEach { file ->
+    files.forEach { file ->
         val (_, textWithTags) = configureEditorByExistingFile(file, project)
         atLeastOneFile = true
         content[file] = textWithTags
     }
     Assert.assertTrue(atLeastOneFile)
-    kotlinFiles.forEach { file ->
+    files.forEach { file ->
         val (editor, _) = configureEditorByExistingFile(file, project, content[file])
         analyzer.checkHighlighting(project, editor)
     }
diff --git a/idea/idea-gradle/tests/org/jetbrains/kotlin/gradle/ImportAndCheckHighlighting.kt b/idea/idea-gradle/tests/org/jetbrains/kotlin/gradle/ImportAndCheckHighlighting.kt
index 6003cfe467b..8850093e556 100644
--- a/idea/idea-gradle/tests/org/jetbrains/kotlin/gradle/ImportAndCheckHighlighting.kt
+++ b/idea/idea-gradle/tests/org/jetbrains/kotlin/gradle/ImportAndCheckHighlighting.kt
@@ -29,16 +29,14 @@ class ImportAndCheckHighlighting : MultiplePluginVersionGradleImportingTestCase(
         importProject()
         val project = myTestFixture.project
         checkFiles(
-            files,
+            files.filter { it.extension == "kt" || it.extension == "java" },
             project,
             object : GradleDaemonAnalyzerTestCase(
                 testLineMarkers = testLineMarkers,
                 checkWarnings = checkWarnings,
                 checkInfos = false
             ) {}
-        ) { file ->
-            file.extension == "kt" || file.extension == "java"
-        }
+        )
     }
 
     override fun testDataDirName(): String {