private static void replaceStatementKeepComments(PsiStatement replacedStatement,
                                                   PsiReturnStatement returnStatement,
                                                   @NotNull ModPsiUpdater updater) {
    updater.highlight(new CommentTracker().replaceAndRestoreComments(replacedStatement, returnStatement));
  }
private static void replaceStatementKeepComments(PsiStatement replacedStatement,
                                                   PsiReturnStatement returnStatement,
                                                   @NotNull ModPsiUpdater updater)