Axolotl move too slowly on cactus/magma block causing them to die

if an axolotl is on a cactus or magma block and there is water near them, they start taking damage so they play dead but in doing so they move very slowly and die, without even trying to go into the water