Fix ConfigurationExtractor compilation finding

1 files changed, 2 insertions, 4 deletions

diff --git a/runners/gradle-plugin/src/main/kotlin/org/jetbrains/dokka/gradle/ConfigurationExtractor.kt b/runners/gradle-plugin/src/main/kotlin/org/jetbrains/dokka/gradle/ConfigurationExtractor.kt
index 393a3077..ca7e4f1e 100644
--- a/runners/gradle-plugin/src/main/kotlin/org/jetbrains/dokka/gradle/ConfigurationExtractor.kt
+++ b/runners/gradle-plugin/src/main/kotlin/org/jetbrains/dokka/gradle/ConfigurationExtractor.kt
@@ -33,11 +33,9 @@ class ConfigurationExtractor(private val project: Project) {
             ?: run { project.logger.error("No source set with name '$sourceSetName' found"); return null }
         val compilation = when (projectExtension) {
             is KotlinMultiplatformExtension -> projectExtension.targets.flatMap { it.compilations }
-                .first { it.kotlinSourceSets.contains(sourceSet) }
+                .find { it.kotlinSourceSets.contains(sourceSet) }
             is KotlinSingleTargetExtension -> projectExtension.target.compilations.find {
-                it.kotlinSourceSets.contains(
-                    sourceSet
-                )
+                it.kotlinSourceSets.contains(sourceSet)
             }
             else -> null
         } ?: run { project.logger.error("No compilation found for set with name '$sourceSetName'"); return null }