To match the opm-autodiff implementation that only handles one phase each time. Polymer equation is handled with the water phase together, since it effects the water phase.