Test water.cave_water_state instead of water.flow

So as to not require water.flow to return @cave_water_state