Fixed newton_raphson args

- Lagrange solver passes state to newton_raphson