While underwater the air bar disappears when your head reaches the top area of a block and appears again when you go higher or lower. It always resets when this happens.

Tested in source blocks and flowing blocks of water.