diff --git a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/state/LLFirResolvableResolveSession.kt b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/state/LLFirResolvableResolveSession.kt index 12fc32c823d..af111b3f842 100644 --- a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/state/LLFirResolvableResolveSession.kt +++ b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/state/LLFirResolvableResolveSession.kt @@ -101,7 +101,7 @@ internal abstract class LLFirResolvableResolveSession( ktDeclaration: KtDeclaration, phase: FirResolvePhase ): FirBasedSymbol<*> { - val module = getModule(ktDeclaration) + val module = getModule(ktDeclaration.originalDeclaration ?: ktDeclaration) retryOnInvalidSession { return when (getModuleKind(module)) { ModuleKind.RESOLVABLE_MODULE -> findSourceFirSymbol(ktDeclaration, module).also { resolveFirToPhase(it.fir, phase) }