Implement correct testing of non-linearity for linear solvers