In my world I have started to breed Axolotls and they have an area which has both water and land for them to wander. If I fly away with my elytra for some period of time (over 5 mins) and come back, any Axolotl on land will start to die. I suspect that this may be because when the chunk is unloaded the Axolotl, if on land, will stay on land until the chunk is loaded again and the 6000 tick timer for them to die when not in water is ongoing regardless whether the chunk is loaded or not.

Steps to Reproduce:

1. Place any axolotl from a bucket.
2. Remove the water the axolotl is in so it is on land.
3. Unload the axolotl for five minutes.
4. Reload the axolotl.
5. Observe the axolotl suffocate.

Observed Results:
The axolotl suffocates and likely dies.

Expected Results:
The suffocation timer of the axolotl remains where it was before the axolotl was unloaded, and the axolotl doesn't suffocate