@Override
  public ProblemDescriptor @Nullable [] checkClass(@NotNull UClass uClass, @NotNull InspectionManager manager, boolean isOnTheFly) {
    PsiClass aClass = uClass.getJavaPsi();
    if (InheritanceUtil.isInheritor(aClass, true, BASE_WALKING_VISITOR_NAME) ||
        InheritanceUtil.isInheritor(aClass, true, BASE_VISITOR_NAME)) {
      if (hasMethod(uClass, "visitReturnStatement", PsiReturnStatement.class.getName())) {
        final boolean visitLambdaMissing = !hasMethod(uClass, "visitLambdaExpression", PsiLambdaExpression.class.getName());
        final boolean visitClassMissing = !hasMethod(uClass, "visitClass", PsiClass.class.getName());
        if (visitLambdaMissing || visitClassMissing) {
          final ProblemsHolder holder = createProblemsHolder(uClass, manager, isOnTheFly);
          ProblemHolderUtilKt.registerUProblem(holder, uClass,
                                               DevKitBundle.message("inspections.unsafe.return.message"),
                                               createFixes(uClass, visitLambdaMissing, visitClassMissing));
          return holder.getResultsArray();
        }
      }
    }
    return ProblemDescriptor.EMPTY_ARRAY;
  }
@Override
  public ProblemDescriptor @Nullable [] checkClass(@NotNull UClass uClass, @NotNull InspectionManager manager, boolean isOnTheFly)