minecaft axolots die when in air

some axolotl will die after 5 min when on land or air
most of my axolotl just died after i puted them in a cage but the only way to make them don't die is putting water near them
also naming an axolotl '' exciton '' they will despawn if you go far