Axolotls sometimes do not play dead and regenerate when their health is low

The bug

Axolotls sometimes do not play dead and regenerate when their health is low.

To reproduce

1. Under water, summon a drowned.
2. Summon an axolotl with low health:

   /summon minecraft:axolotl ~ ~ ~ {Health:2.0f}
   

Expected result

The axolotl would play dead and regenerate.

Actual result

The axolotl tries to attack the drowned, which retaliates and kills the axolotl immediately.

Videos

https://youtu.be/8dLFdczSFGU shorter version
https://youtu.be/ASQVz3CW2pY longer version