Axolotl will try to approach the water source surrounded by blocks in different positions.
In doing so, it may also swim through the air when it in the water.

 It is difficult to explain this situation in writing, so please watch the attached video.