void (PsiExpressionStatement expressionStatement) { CommentTracker tracker = new CommentTracker(); String valueKeyword = PsiKeyword.YIELD; tracker.replaceAndRestoreComments(expressionStatement, "{ " + valueKeyword + " " + tracker.text(expressionStatement) + "\n }"); }