diff --git a/idea/kotlin-gradle-tooling/src/KotlinGradleModelBuilder.kt b/idea/kotlin-gradle-tooling/src/KotlinGradleModelBuilder.kt
index f6db0358ce6..ab1cd4fdb02 100644
--- a/idea/kotlin-gradle-tooling/src/KotlinGradleModelBuilder.kt
+++ b/idea/kotlin-gradle-tooling/src/KotlinGradleModelBuilder.kt
@@ -165,7 +165,7 @@ class KotlinGradleModelBuilder : ModelBuilderService {
 
     private fun Task.getSourceSetName(): String {
         return try {
-            javaClass.methods.firstOrNull { it.name.startsWith("getSourceSetName") && it.parameterCount == 0 }?.invoke(this) as? String
+            javaClass.methods.firstOrNull { it.name.startsWith("getSourceSetName") && it.parameterTypes.isEmpty() }?.invoke(this) as? String
         } catch (e : InvocationTargetException) {
             null // can be thrown if property is not initialized yet
         } ?: "main"