Axolotls can move while regenerating

When an axolotl is regenerating and you pick it up while its regenerating and place it back it can move while regenerating

Steps to reproduce [by [Mod] EVGENSYPERPRO]:

1. Get the axolotl to regenerate.
2. Catch him in a bucket.
3. Free the axolotl.

Observed results:

The axolotl will regenerate and swim.

Expected results:

When the axolotl regenerates, it should not move.

Notes:

In the Java version, when the axolotl is released from the bucket, the regeneration effect disappears MC-228903.

Video:

Minecraft 2021-06-13 20-32-25.mp4