diff --git a/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/FirModuleResolveState.kt b/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/FirModuleResolveState.kt
index be2d1616840..df922514fb6 100644
--- a/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/FirModuleResolveState.kt
+++ b/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/FirModuleResolveState.kt
@@ -100,16 +100,18 @@ internal class FirModuleResolveStateImpl(
         toPhase: FirResolvePhase,
         towerDataContextCollector: FirTowerDataContextCollector
     ) {
-        firLazyDeclarationResolver.runLazyResolveWithoutLock(
-            firFunction,
-            rootModuleSession.cache,
-            containerFirFile,
-            firIdeProvider,
-            fromPhase = firFunction.resolvePhase,
-            toPhase,
-            towerDataContextCollector,
-            checkPCE = false
-        )
+        firFileBuilder.runCustomResolveWithPCECheck(containerFirFile, rootModuleSession.cache) {
+            firLazyDeclarationResolver.runLazyResolveWithoutLock(
+                firFunction,
+                rootModuleSession.cache,
+                containerFirFile,
+                firIdeProvider,
+                fromPhase = firFunction.resolvePhase,
+                toPhase,
+                towerDataContextCollector,
+                checkPCE = true
+            )
+        }
     }
 }