Update non-linearity checks so that linear solvers pass them