diff --git a/compiler/frontend/src/org/jetbrains/kotlin/cfg/cfgContainingDeclarationUtils.kt b/compiler/frontend/src/org/jetbrains/kotlin/cfg/cfgContainingDeclarationUtils.kt
index 49a8412feff..2248219c118 100644
--- a/compiler/frontend/src/org/jetbrains/kotlin/cfg/cfgContainingDeclarationUtils.kt
+++ b/compiler/frontend/src/org/jetbrains/kotlin/cfg/cfgContainingDeclarationUtils.kt
@@ -18,9 +18,10 @@ val KtElement.containingDeclarationForPseudocode: KtDeclaration?
     get() = getParentOfType(this, KtDeclarationWithBody::class.java, KtClassOrObject::class.java, KtScript::class.java)
         ?: getNonStrictParentOfType<KtProperty>()
 
-// Should return KtDeclarationWithBody, KtClassOrObject, or KtClassInitializer
+// Should return KtDeclarationWithBody, KtClassOrObject, KtClassInitializer or KtScriptInitializer
 fun KtElement.getElementParentDeclaration(): KtDeclaration? =
-    getParentOfType(this, KtDeclarationWithBody::class.java, KtClassOrObject::class.java, KtClassInitializer::class.java)
+    getParentOfType(this, KtDeclarationWithBody::class.java, KtClassOrObject::class.java, KtClassInitializer::class.java,
+        KtScriptInitializer::class.java)
 
 fun KtDeclaration?.getDeclarationDescriptorIncludingConstructors(context: BindingContext): DeclarationDescriptor? {
     val descriptor = context.get(DECLARATION_TO_DESCRIPTOR, (this as? KtClassInitializer)?.containingDeclaration ?: this)