diff --git a/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/util/declarationUtils.kt b/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/util/declarationUtils.kt
index 6d33a71c210..fe2fffbf6fc 100644
--- a/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/util/declarationUtils.kt
+++ b/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/util/declarationUtils.kt
@@ -91,7 +91,8 @@ private fun KtDeclaration.findSourceNonLocalFirDeclarationByProvider(
                 firFile.declarations
             }
             val original = originalDeclaration
-            declarations?.first { it.psi === this || it.psi == original }
+            declarations?.firstOrNull { it.psi == this || it.psi == original }
+                ?: error("Cannot find corresponding fir for\n${this.getElementTextInContext()}")
         }
         this is KtConstructor<*> -> {
             val containingClass = containingClassOrObject