Small adjustments to the new model

2 files changed, 3 insertions, 5 deletions

diff --git a/plugins/base/src/test/kotlin/markdown/KDocTest.kt b/plugins/base/src/test/kotlin/markdown/KDocTest.kt
index c58c4e30..5099c46f 100644
--- a/plugins/base/src/test/kotlin/markdown/KDocTest.kt
+++ b/plugins/base/src/test/kotlin/markdown/KDocTest.kt
@@ -27,9 +27,8 @@ open class KDocTest : AbstractCoreTest() {
 
     private fun actualDocumentationNode(modulePageNode: ModulePageNode) =
         (modulePageNode.documentable?.children?.first() as Package)
-            .classlikes.first()
-            .platformInfo.first()
-            .documentationNode
+            .classlikes.single()
+            .documentation.values.single()
 
 
     protected fun executeTest(kdoc: String, expectedDocumentationNode: DocumentationNode) {
diff --git a/plugins/base/src/test/kotlin/multiplatform/BasicMultiplatformTest.kt b/plugins/base/src/test/kotlin/multiplatform/BasicMultiplatformTest.kt
index f9431bbb..ae5b4e06 100644
--- a/plugins/base/src/test/kotlin/multiplatform/BasicMultiplatformTest.kt
+++ b/plugins/base/src/test/kotlin/multiplatform/BasicMultiplatformTest.kt
@@ -47,8 +47,7 @@ class BasicMultiplatformTest : AbstractCoreTest() {
             configuration
         ) {
             pagesGenerationStage = {
-                println(it.dri)
-                assertEquals(7, it.parentMap.size)
+                assertEquals(6, it.parentMap.size)
             }
         }
     }