Glitchy Axolotl

So spawning a axolotl in 1 block high water causes it to glitch out. Its looks kind of like there doing a head bob but incredibly fast. another glitch I found is if you get a axolotl on a lead,  the axolotl will swim incredibly fast in the water and swim on land around the player if you take them out of the water... I'm starting to think they don't like to be tied up...