Axolot playing dead animation don't reset correctly.

Hello i found a animation bug of axolotl mob animation when this mod stop to playing dead he is tile forever and he don't return at his normal position,please fix this bug bug for 1.17 part 1.