void (@NotNull ActionContext context, @NotNull PsiExpression expression, @NotNull ModPsiUpdater updater) { CommentTracker tracker = new CommentTracker(); tracker.replaceAndRestoreComments(expression.getParent(), "return " + tracker.text(expression) + ";"); }