void replaceContinueWithReturn(PsiElementFactory factory) {
    PsiLoopStatement currentLoop = PsiTreeUtil.getParentOfType(myStatements[0], PsiLoopStatement.class);
    if (currentLoop == null) return;
    for (int i = 0, length = myStatements.length; i < length; i++) {
      PsiStatement statement = myStatements[i];
      if(statement instanceof PsiContinueStatement) {
        myStatements[i] = factory.createStatementFromText("return;", null);
        continue;
      }
      StreamEx.ofTree(statement, (PsiElement s) -> StreamEx.of(s.getChildren()))
        .select(PsiContinueStatement.class)
        .filter(stmt -> stmt.findContinuedStatement() == currentLoop)
        .forEach(stmt -> new CommentTracker().replaceAndRestoreComments(stmt, "return;"));
    }
  }
void replaceContinueWithReturn(PsiElementFactory factory)