3 files changed, 5 insertions, 4 deletions
diff --git a/plugins/base/src/test/kotlin/basic/DokkaBasicTests.kt b/plugins/base/src/test/kotlin/basic/DokkaBasicTests.kt
index c1044e09..c9f4b156 100644
--- a/plugins/base/src/test/kotlin/basic/DokkaBasicTests.kt
+++ b/plugins/base/src/test/kotlin/basic/DokkaBasicTests.kt
@@ -31,8 +31,9 @@ class DokkaBasicTests : AbstractCoreTest() {
             configuration
         ) {
             pagesGenerationStage = {
-                println(it.dri)
-                assertTrue(it.getClasslikeToMemberMap().filterKeys { it.name == "Test" }.entries.firstOrNull()?.value?.size == 2)
+                val root = it as ModulePageNode
+                println(root.dri)
+                assertTrue(root.getClasslikeToMemberMap().filterKeys { root.name == "Test" }.entries.firstOrNull()?.value?.size == 2)
             }
         }
     }
diff --git a/plugins/base/src/test/kotlin/enums/EnumsTest.kt b/plugins/base/src/test/kotlin/enums/EnumsTest.kt
index a8b78baa..125e37e2 100644
--- a/plugins/base/src/test/kotlin/enums/EnumsTest.kt
+++ b/plugins/base/src/test/kotlin/enums/EnumsTest.kt
@@ -231,7 +231,7 @@ class EnumsTest : AbstractCoreTest() {
     }
 
 
-    fun ModulePageNode.getClasslikeToMemberMap() =
+    fun RootPageNode.getClasslikeToMemberMap() =
         this.parentMap.filterValues { it is ClasslikePageNode }.entries.groupBy({ it.value }) { it.key }
 
     private fun ContentGroup.constructorSignature(): String =
diff --git a/plugins/base/src/test/kotlin/markdown/KDocTest.kt b/plugins/base/src/test/kotlin/markdown/KDocTest.kt
index f9d717b0..218f7244 100644
--- a/plugins/base/src/test/kotlin/markdown/KDocTest.kt
+++ b/plugins/base/src/test/kotlin/markdown/KDocTest.kt
@@ -39,7 +39,7 @@ abstract class KDocTest : AbstractCoreTest() {
             pagesGenerationStage = {
                 assertEquals(
                     expectedDocumentationNode,
-                    actualDocumentationNode(it)
+                    actualDocumentationNode(it as ModulePageNode)
                 )
             }
         }