diff --git a/plugins/kotlin/highlighting/highlighting-minimal/src/org/jetbrains/kotlin/idea/highlighter/AbstractHighlightingPassBase.kt b/plugins/kotlin/highlighting/highlighting-minimal/src/org/jetbrains/kotlin/idea/highlighter/AbstractHighlightingPassBase.kt
index 2db68b5ea824..ed0a276f0f47 100644
--- a/plugins/kotlin/highlighting/highlighting-minimal/src/org/jetbrains/kotlin/idea/highlighter/AbstractHighlightingPassBase.kt
+++ b/plugins/kotlin/highlighting/highlighting-minimal/src/org/jetbrains/kotlin/idea/highlighter/AbstractHighlightingPassBase.kt
@@ -24,7 +24,12 @@ abstract class AbstractHighlightingPassBase(
 
     override fun doCollectInformation(progress: ProgressIndicator) {
         val holder = HighlightInfoHolder(file)
-        runAnnotatorWithContext(file, holder)
+        if (IGNORE_IN_TESTS) {
+            assert(ApplicationManager.getApplication().isUnitTestMode)
+        }
+        else {
+            runAnnotatorWithContext(file, holder)
+        }
         applyInformationInBackground(holder)
     }
 
@@ -55,10 +60,6 @@ abstract class AbstractHighlightingPassBase(
     }
 
     private fun applyInformationInBackground(holder: HighlightInfoHolder) {
-        if (IGNORE_IN_TESTS) {
-            assert(ApplicationManager.getApplication().isUnitTestMode)
-            return
-        }
         val result:MutableList<HighlightInfo> = ArrayList(holder.size())
         for (i in 0 until holder.size()) {
             result.add(holder.get(i))