From 8f70ef41f0faa7aabb5a43ac9ec787c6a396c54e Mon Sep 17 00:00:00 2001
From: Ilya Kirillov <ilya.kirillov@jetbrains.com>
Date: Mon, 21 Jun 2021 10:18:05 +0200
Subject: [PATCH] FIR IDE: consider resolve inside completion candidate checker

---
 .../api/fir/components/KtFirCompletionCandidateChecker.kt     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/frontend/api/fir/components/KtFirCompletionCandidateChecker.kt b/idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/frontend/api/fir/components/KtFirCompletionCandidateChecker.kt
index 028dbdf196c..9d311871db0 100644
--- a/idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/frontend/api/fir/components/KtFirCompletionCandidateChecker.kt
+++ b/idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/frontend/api/fir/components/KtFirCompletionCandidateChecker.kt
@@ -42,7 +42,9 @@ internal class KtFirCompletionCandidateChecker(
         possibleExplicitReceiver: KtExpression?,
     ): KtExtensionApplicabilityResult = withValidityAssertion {
         require(firSymbolForCandidate is KtFirSymbol<*>)
-        return firSymbolForCandidate.firRef.withFir(phase = FirResolvePhase.IMPLICIT_TYPES_BODY_RESOLVE) { declaration ->
+        return firSymbolForCandidate.firRef.withFirWithPossibleResolveInside(
+            phase = FirResolvePhase.IMPLICIT_TYPES_BODY_RESOLVE
+        ) { declaration ->
             check(declaration is FirCallableDeclaration<*>)
             checkExtension(declaration, originalFile, nameExpression, possibleExplicitReceiver)
         }