diff --git a/idea/tests/org/jetbrains/kotlin/idea/structureView/KotlinFileStructureTestBase.kt b/idea/tests/org/jetbrains/kotlin/idea/structureView/KotlinFileStructureTestBase.kt
index 051a7c87e37..4d4e55912b0 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/structureView/KotlinFileStructureTestBase.kt
+++ b/idea/tests/org/jetbrains/kotlin/idea/structureView/KotlinFileStructureTestBase.kt
@@ -76,8 +76,9 @@ abstract class KotlinFileStructureTestBase : KotlinLightCodeInsightFixtureTestCa
 
     protected fun checkResult() {
         val printInfo = Queryable.PrintInfo(arrayOf("text"), arrayOf("location"))
-        TreeUtil.expandAll(popupFixture.tree);
-        val popupText = StructureViewUtil.print(popupFixture.tree, false, printInfo, null).trim { it <= ' ' }
-        UsefulTestCase.assertSameLinesWithFile(testDataPath + "/" + treeFileName, popupText)
+        TreeUtil.expandAll(popupFixture.tree) {
+            val popupText = StructureViewUtil.print(popupFixture.tree, false, printInfo, null).trim { it <= ' ' }
+            UsefulTestCase.assertSameLinesWithFile(testDataPath + "/" + treeFileName, popupText)
+        }
     }
 }
\ No newline at end of file