diff --git a/compiler/psi/src/org/jetbrains/kotlin/psi/KtCodeFragment.kt b/compiler/psi/src/org/jetbrains/kotlin/psi/KtCodeFragment.kt
index dc54dcb17fc..27345992b08 100644
--- a/compiler/psi/src/org/jetbrains/kotlin/psi/KtCodeFragment.kt
+++ b/compiler/psi/src/org/jetbrains/kotlin/psi/KtCodeFragment.kt
@@ -172,7 +172,7 @@ abstract class KtCodeFragment(
     override fun getExceptionHandler() = exceptionHandler
 
     fun getContextContainingFile(): KtFile? {
-        return getOriginalContext()?.containingKtFile
+        return getOriginalContext()?.takeIf { it.isValid }?.containingKtFile
     }
 
     fun getOriginalContext(): KtElement? {