diff --git a/idea/idea-analysis/src/org/jetbrains/kotlin/idea/highlighter/DebugInfoAnnotator.kt b/idea/idea-analysis/src/org/jetbrains/kotlin/idea/highlighter/DebugInfoAnnotator.kt
index cae41ef0d0a..4ca582e36e6 100644
--- a/idea/idea-analysis/src/org/jetbrains/kotlin/idea/highlighter/DebugInfoAnnotator.kt
+++ b/idea/idea-analysis/src/org/jetbrains/kotlin/idea/highlighter/DebugInfoAnnotator.kt
@@ -62,7 +62,8 @@ class DebugInfoAnnotator : Annotator {
 
     companion object {
         val isDebugInfoEnabled: Boolean
-            get() = ApplicationManager.getApplication()
-                .isInternal && (KotlinPluginUtil.isSnapshotVersion() || KotlinPluginUtil.isDevVersion())
+            get() = ApplicationManager.getApplication().isUnitTestMode ||
+                    (ApplicationManager.getApplication().isInternal &&
+                            (KotlinPluginUtil.isSnapshotVersion() || KotlinPluginUtil.isDevVersion()))
     }
 }