diff --git a/orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt b/orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt index b4e37d8b..7cf88a25 100644 --- a/orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt +++ b/orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt @@ -43,11 +43,13 @@ class LambdaExtractor(val ruleNames: List, val lambdaName: String) : Kot var result: String? = null override fun enterAnnotatedLambda(ctx: KotlinParser.AnnotatedLambdaContext?) { - val puec = ctx!!.parent!!.parent!!.parent!! as KotlinParser.PostfixUnaryExpressionContext - val identifier = puec.primaryExpression().simpleIdentifier().Identifier().text - if (identifier == lambdaName) { - if (result == null) { - result = ctx.verbatimText(1, 1) + val puec = ctx?.parent?.parent?.parent as? KotlinParser.PostfixUnaryExpressionContext + if (puec != null) { + val identifier = puec.primaryExpression()?.simpleIdentifier()?.Identifier()?.text + if (identifier == lambdaName) { + if (result == null) { + result = ctx.verbatimText(1, 1) + } } } }