diff --git a/idea/idea-fir/src/org/jetbrains/kotlin/idea/fir/highlighter/KotlinFirPsiChecker.kt b/idea/idea-fir/src/org/jetbrains/kotlin/idea/fir/highlighter/KotlinFirPsiChecker.kt
index efb79cf0e78..43557522518 100644
--- a/idea/idea-fir/src/org/jetbrains/kotlin/idea/fir/highlighter/KotlinFirPsiChecker.kt
+++ b/idea/idea-fir/src/org/jetbrains/kotlin/idea/fir/highlighter/KotlinFirPsiChecker.kt
@@ -30,7 +30,7 @@ class KotlinFirPsiChecker : AbstractKotlinPsiChecker() {
             throw ProcessCanceledException()
         }
         analyze(element) {
-            highlightDiagnostics(element, this, holder)
+            this.highlightDiagnostics(element, holder)
 
             FirAfterResolveHighlightingVisitor
                 .createListOfVisitors(this, holder)
@@ -38,7 +38,7 @@ class KotlinFirPsiChecker : AbstractKotlinPsiChecker() {
         }
     }
 
-    private fun highlightDiagnostics(element: KtElement, analysisSession: KtAnalysisSession, holder: AnnotationHolder) = with(analysisSession) {
+    private fun KtAnalysisSession.highlightDiagnostics(element: KtElement, holder: AnnotationHolder) {
         val diagnostics = element.getDiagnostics()
         if (diagnostics.isEmpty()) return