Underwater visibility resets when re-entering world

How to reproduce

1. Go to a ≥2 block deep body of water

2. Wait until underwater area is visible (theoretically easier with night vision and water breathing)

3. When underwater area is visible, leave the world

4. Re-enter the world

Underwater visibility should remain when logging back into a world (at least that's what I expected).

Code analysis

Code analysis by @unknown can be found in this comment.