diff --git a/libraries/tools/kotlin-gradle-plugin-core/src/main/kotlin/org/jetbrains/kotlin/gradle/tasks/Tasks.kt b/libraries/tools/kotlin-gradle-plugin-core/src/main/kotlin/org/jetbrains/kotlin/gradle/tasks/Tasks.kt
index 8f4e15c72c9..708e302d89c 100644
--- a/libraries/tools/kotlin-gradle-plugin-core/src/main/kotlin/org/jetbrains/kotlin/gradle/tasks/Tasks.kt
+++ b/libraries/tools/kotlin-gradle-plugin-core/src/main/kotlin/org/jetbrains/kotlin/gradle/tasks/Tasks.kt
@@ -104,12 +104,8 @@ abstract class AbstractKotlinCompile<T : CommonCompilerArguments>() : AbstractCo
 
     private fun getKotlinSources(): List<File> = (getSource() as Iterable<File>).filter { it.isKotlinFile() }
 
-    protected fun File.isKotlinFile(): Boolean {
-        return when (FilenameUtils.getExtension(name).toLowerCase()) {
-            "kt", "kts" -> true
-            else -> false
-        }
-    }
+    protected fun File.isKotlinFile(): Boolean =
+            FilenameUtils.isExtension(name.toLowerCase(), listOf("kt", "kts"))
 
     private fun populateSources(args:T, sources: List<File>) {
         args.freeArgs = sources.map { it.absolutePath }