axolotl does not work

When you take them out of the water and put them on land, the axolotls do not look for a way to go to the water, they stay on land (the oceans give lag due to the fish killed by the axolotls)