From 61f6db87c40ffae9e4a82b70840a42ae1a224dd7 Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Tue, 26 Sep 2017 16:39:22 +0200
Subject: [PATCH] Use convertOpt() for getting inner classes of script

---
 .../src/org/jetbrains/uast/kotlin/declarations/KotlinUClass.kt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUClass.kt b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUClass.kt
index 7eecbbfd959..3e9e0367035 100644
--- a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUClass.kt
+++ b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUClass.kt
@@ -157,7 +157,8 @@ class KotlinScriptUClass(
 
     override fun getInitializers(): Array<UClassInitializer> = super.getInitializers()
 
-    override fun getInnerClasses(): Array<UClass> = super.getInnerClasses()
+    override fun getInnerClasses(): Array<UClass> =
+            psi.innerClasses.mapNotNull { getLanguagePlugin().convertOpt<UClass>(it, this) }.toTypedArray()
 
     override fun getMethods(): Array<UMethod> = psi.methods.map(this::createUMethod).toTypedArray()