void (@NotNull ReturnContext context) { Set<PsiElement> skippedEmptyStatements = new HashSet<>(); getPrevNonEmptyStatement(context.returnStatement, skippedEmptyStatements); skippedEmptyStatements.forEach(PsiElement::delete); removeElementKeepComments(context.returnStatement); }