diff --git a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
index 8c1cb1c0e32..eb726bdd07b 100644
--- a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
+++ b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
@@ -332,6 +332,14 @@ private class ContextCollectorVisitor(
         }
     }
 
+    override fun visitCodeFragment(codeFragment: FirCodeFragment) {
+        codeFragment.lazyResolveToPhase(FirResolvePhase.BODY_RESOLVE)
+
+        context.withCodeFragment(codeFragment, holder) {
+            super.visitCodeFragment(codeFragment)
+        }
+    }
+
     override fun visitAnnotationCall(annotationCall: FirAnnotationCall) {
         dumpContext(annotationCall, ContextKind.SELF)