1 files changed, 2 insertions, 2 deletions
diff --git a/testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt b/testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt
index 3fde41a0..ee62d8c7 100644
--- a/testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt
+++ b/testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt
@@ -112,7 +112,7 @@ abstract class AbstractCoreTest {
         var documentablesFirstTransformationStep: (List<DModule>) -> Unit = {}
         var documentablesMergingStage: (DModule) -> Unit = {}
         var documentablesTransformationStage: (DModule) -> Unit = {}
-        var pagesGenerationStage: (ModulePageNode) -> Unit = {}
+        var pagesGenerationStage: (RootPageNode) -> Unit = {}
         var pagesTransformationStage: (RootPageNode) -> Unit = {}
         var renderingStage: (RootPageNode, DokkaContext) -> Unit = { a, b -> }
 
@@ -246,7 +246,7 @@ data class TestMethods(
     val documentablesFirstTransformationStep: (List<DModule>) -> Unit,
     val documentablesMergingStage: (DModule) -> Unit,
     val documentablesTransformationStage: (DModule) -> Unit,
-    val pagesGenerationStage: (ModulePageNode) -> Unit,
+    val pagesGenerationStage: (RootPageNode) -> Unit,
     val pagesTransformationStage: (RootPageNode) -> Unit,
     val renderingStage: (RootPageNode, DokkaContext) -> Unit
 )
 \ No newline at end of file