Axolotl's head getting stuck on the ground

I spawned an axolotl, so it tried to reach the nearest water source. But it's head started to go in the ground while it was heading towards the water source. The water's Y level was lower than the axolotl's Y level.