diff --git a/idea/src/org/jetbrains/kotlin/idea/codeInliner/CodeToInlineBuilder.kt b/idea/src/org/jetbrains/kotlin/idea/codeInliner/CodeToInlineBuilder.kt
index ee82bef4d57..43447728367 100644
--- a/idea/src/org/jetbrains/kotlin/idea/codeInliner/CodeToInlineBuilder.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/codeInliner/CodeToInlineBuilder.kt
@@ -64,8 +64,7 @@ class CodeToInlineBuilder(
 
         processReferences(codeToInline, bindingContext)
 
-        // NB: we must check in codeToInline, otherwise we get AE from addPostInsertionAction
-        if (mainExpression != null && mainExpression in codeToInline) {
+        if (mainExpression != null) {
             val functionLiteralExpression = mainExpression.unpackFunctionLiteral(true)
             if (functionLiteralExpression != null) {
                 val functionLiteralParameterTypes = getParametersForFunctionLiteral(functionLiteralExpression, bindingContext)