1 files changed, 4 insertions, 3 deletions
diff --git a/plugins/kotlin/fir/src/org/jetbrains/kotlin/idea/fir/highlighter/visitors/FirAfterResolveHighlightingVisitor.kt b/plugins/kotlin/fir/src/org/jetbrains/kotlin/idea/fir/highlighter/visitors/FirAfterResolveHighlightingVisitor.kt
index b7d71d9d4720..ff64768728bb 100644
--- a/plugins/kotlin/fir/src/org/jetbrains/kotlin/idea/fir/highlighter/visitors/FirAfterResolveHighlightingVisitor.kt
+++ b/plugins/kotlin/fir/src/org/jetbrains/kotlin/idea/fir/highlighter/visitors/FirAfterResolveHighlightingVisitor.kt
@@ -3,9 +3,12 @@
 package org.jetbrains.kotlin.idea.fir.highlighter.visitors
 
 import com.intellij.lang.annotation.AnnotationHolder
+import com.intellij.lang.annotation.HighlightSeverity
 import com.intellij.openapi.editor.colors.TextAttributesKey
 import com.intellij.openapi.util.TextRange
+import com.intellij.util.applyIf
 import org.jetbrains.kotlin.analysis.api.KtAnalysisSession
+import org.jetbrains.kotlin.idea.fir.highlighter.HiglightingFactory
 import org.jetbrains.kotlin.idea.highlighter.AbstractHighlightingVisitor
 
 abstract class FirAfterResolveHighlightingVisitor(
@@ -14,9 +17,7 @@ abstract class FirAfterResolveHighlightingVisitor(
 ) : AbstractHighlightingVisitor() {
 
     override fun createInfoAnnotation(textRange: TextRange, message: String?, textAttributes: TextAttributesKey?) {
-        // TODO: Temporary use deprecated for FIR plugin as it is supposes to be rewritten fully
-        holder.createInfoAnnotation(textRange, message)
-            .also { annotation -> textAttributes?.let { annotation.textAttributes = textAttributes } }
+        HiglightingFactory.createInfoAnnotation(holder, textRange, message, textAttributes)
     }
 
     companion object {