Always return a number from the material laws. Note that this is not a regularization. It just prevents the function from returning NaN. In the case of sw = 0.0 we still return inf as it is the natural limit of the equation. Also, the linear law can e.g. still return unphysical values as they are still valid numbers and in the range of the function.
Also apply on next.