diff --git a/idea/tests/org/jetbrains/kotlin/idea/folding/FoldingAfterOptimizeImportsTest.kt b/idea/tests/org/jetbrains/kotlin/idea/folding/FoldingAfterOptimizeImportsTest.kt
index 6d9c3b8340a..c1d77e09b99 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/folding/FoldingAfterOptimizeImportsTest.kt
+++ b/idea/tests/org/jetbrains/kotlin/idea/folding/FoldingAfterOptimizeImportsTest.kt
@@ -32,9 +32,6 @@ class FoldingAfterOptimizeImportsTest : AbstractKotlinFoldingTest() {
     private val fixture: JavaCodeInsightTestFixture
         get() = myFixture!!
 
-    private val editor: Editor
-        get() = fixture.getEditor()!!
-
     private val fileText: String
         get() = fixture.getFile()!!.getText()!!
 
@@ -51,7 +48,7 @@ class FoldingAfterOptimizeImportsTest : AbstractKotlinFoldingTest() {
 
         doTestWithSettings(fileText) {
             fileText ->
-            CodeFoldingManager.getInstance(fixture.getProject())!!.buildInitialFoldings(editor)
+            CodeFoldingManager.getInstance(fixture.getProject())!!.buildInitialFoldings(getEditor())
             getFoldingRegion(0).checkRegion(false, findStringWithPrefixes("// REGION BEFORE: "))
 
             CommandProcessor.getInstance()?.executeCommand(fixture.getProject(),
@@ -65,7 +62,7 @@ class FoldingAfterOptimizeImportsTest : AbstractKotlinFoldingTest() {
 
     private fun getFoldingRegion(number: Int): FoldRegion {
         fixture.doHighlighting()
-        val model = editor.getFoldingModel()
+        val model = getEditor().getFoldingModel()
         val foldingRegions = model.getAllFoldRegions()
         assert(foldingRegions.size >= number) { "There is no enough folding regions in file: in file - ${foldingRegions.size} , expected = ${number}" }
         return foldingRegions[number]