void (@NotNull ActionContext context, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) { PsiMethodCallExpression call = PsiTreeUtil.getParentOfType(PsiTreeUtil.getParentOfType(element, PsiLambdaExpression.class), PsiMethodCallExpression.class); SimplifyForEachContext simplifyForEachContext = SimplifyForEachContext.from(call); if (simplifyForEachContext != null) { PsiElement result = simplifyForEachContext.migrate(); if (result != null) { MigrateToStreamFix.simplify(context.project(), result); } } }