diff --git a/compiler/ir/backend.common/src/org/jetbrains/kotlin/KtDiagnosticReporterWithImplicitIrBasedContext.kt b/compiler/ir/backend.common/src/org/jetbrains/kotlin/KtDiagnosticReporterWithImplicitIrBasedContext.kt
index 801b26ba03a..99a0869ddc3 100644
--- a/compiler/ir/backend.common/src/org/jetbrains/kotlin/KtDiagnosticReporterWithImplicitIrBasedContext.kt
+++ b/compiler/ir/backend.common/src/org/jetbrains/kotlin/KtDiagnosticReporterWithImplicitIrBasedContext.kt
@@ -46,7 +46,7 @@ class KtDiagnosticReporterWithImplicitIrBasedContext(
     override fun at(irElement: IrElement, containingIrFile: IrFile): DiagnosticContextImpl =
         at(irElement.toSourceElement(containingIrFile), irElement, containingIrFile)
 
-    fun at(
+    override fun at(
         sourceElement: AbstractKtSourceElement?,
         irElement: IrElement,
         containingFile: IrFile
diff --git a/compiler/ir/ir.tree/src/org/jetbrains/kotlin/ir/IrDiagnosticReporter.kt b/compiler/ir/ir.tree/src/org/jetbrains/kotlin/ir/IrDiagnosticReporter.kt
index d088904da65..b09b854df31 100644
--- a/compiler/ir/ir.tree/src/org/jetbrains/kotlin/ir/IrDiagnosticReporter.kt
+++ b/compiler/ir/ir.tree/src/org/jetbrains/kotlin/ir/IrDiagnosticReporter.kt
@@ -5,6 +5,7 @@
 
 package org.jetbrains.kotlin.ir
 
+import org.jetbrains.kotlin.AbstractKtSourceElement
 import org.jetbrains.kotlin.config.LanguageVersionSettings
 import org.jetbrains.kotlin.diagnostics.KtDiagnosticReporterWithContext.DiagnosticContextImpl
 import org.jetbrains.kotlin.diagnostics.rendering.Renderer
@@ -19,6 +20,7 @@ interface IrDiagnosticReporter {
     fun at(irDeclaration: IrDeclaration): DiagnosticContextImpl
     fun at(irElement: IrElement, containingIrFile: IrFile): DiagnosticContextImpl
     fun at(irElement: IrElement, containingIrDeclaration: IrDeclaration): DiagnosticContextImpl
+    fun at(sourceElement: AbstractKtSourceElement?, irElement: IrElement, containingFile: IrFile): DiagnosticContextImpl
 }
 
 object IrDiagnosticRenderers {