@Override
  protected void invoke(@NotNull ActionContext context, @NotNull PsiParameterListOwner method, @NotNull ModPsiUpdater updater) {
    if (invokeSingleExpressionLambdaFix(method)) {
      return;
    }
    String value = suggestReturnValue(method);
    PsiElementFactory factory = JavaPsiFacade.getElementFactory(context.project());
    PsiReturnStatement returnStatement = (PsiReturnStatement)factory.createStatementFromText("return " + value + ";", method);
    PsiCodeBlock body = ObjectUtils.tryCast(method.getBody(), PsiCodeBlock.class);
    assert body != null;
    returnStatement = (PsiReturnStatement) body.addBefore(returnStatement, body.getRBrace());

    updater.select(Objects.requireNonNull(returnStatement.getReturnValue()));
  }
@Override
  protected void invoke(@NotNull ActionContext context, @NotNull PsiParameterListOwner method, @NotNull ModPsiUpdater updater)