diff --git a/kotlin-analysis/src/main/kotlin/org/jetbrains/dokka/analysis/KotlinAnalysis.kt b/kotlin-analysis/src/main/kotlin/org/jetbrains/dokka/analysis/KotlinAnalysis.kt
index 46557c75f0..64a583b62a 100644
--- a/kotlin-analysis/src/main/kotlin/org/jetbrains/dokka/analysis/KotlinAnalysis.kt
+++ b/kotlin-analysis/src/main/kotlin/org/jetbrains/dokka/analysis/KotlinAnalysis.kt
@@ -83,19 +83,14 @@ abstract class KotlinAnalysis(
 ) : Closeable {
 
     operator fun get(key: DokkaSourceSet): AnalysisContext {
+        return get(key.sourceSetID)
+    }
+    operator fun get(key: DokkaSourceSetID): AnalysisContext {
         return find(key)
             ?: parent?.get(key)
             ?: throw IllegalStateException("Missing EnvironmentAndFacade for sourceSet ${key}")
     }
-
-    operator fun get(sourceSetID: DokkaSourceSetID): AnalysisContext {
-        return find(sourceSetID)
-            ?: parent?.get(sourceSetID)
-            ?: throw IllegalStateException("Missing EnvironmentAndFacade for sourceSet ${sourceSetID.sourceSetName}")
-    }
-
     protected abstract fun find(sourceSetID: DokkaSourceSetID): AnalysisContext?
-    protected abstract fun find(sourceSet: DokkaSourceSet): AnalysisContext?
 }
 
 internal open class EnvironmentKotlinAnalysis(
@@ -103,13 +98,8 @@ internal open class EnvironmentKotlinAnalysis(
     parent: KotlinAnalysis? = null,
 ) : KotlinAnalysis(parent = parent) {
 
-    override fun find(sourceSetID: DokkaSourceSetID): AnalysisContext? {
-        return environments.entries.firstOrNull { (sourceSet, _) -> sourceSet.sourceSetID == sourceSetID }?.value
-    }
-
-    override fun find(sourceSet: DokkaSourceSet): AnalysisContext? {
-        return environments[sourceSet]
-    }
+    override fun find(sourceSetID: DokkaSourceSetID): AnalysisContext? =
+        environments.entries.firstOrNull { (sourceSet, _) -> sourceSet.sourceSetID == sourceSetID }?.value
 
     override fun close() {
         environments.values.forEach(AnalysisContext::close)
diff --git a/plugins/base/src/test/kotlin/model/InheritorsTest.kt b/plugins/base/src/test/kotlin/model/InheritorsTest.kt
index 48ce48c852..1be21afa02 100644
--- a/plugins/base/src/test/kotlin/model/InheritorsTest.kt
+++ b/plugins/base/src/test/kotlin/model/InheritorsTest.kt
@@ -61,12 +61,12 @@ class InheritorsTest : AbstractModelTest("/src/main/kotlin/inheritors/Test.kt",
         val configuration = dokkaConfiguration {
             sourceSets {
                 sourceSet {
-                    displayName = "jvm"
+                    name = "jvm"
                     sourceRoots = listOf("common/src/", "jvm/src/")
                     analysisPlatform = "jvm"
                 }
                 sourceSet {
-                    displayName = "js"
+                    name = "js"
                     sourceRoots = listOf("common/src/", "js/src/")
                     analysisPlatform = "js"
                 }