Attack gets cancelled when diving underwater

This bug occurs with Spinosaurus and Deinocheirus. When an attack is started and the dinosaur dives underwater before the attack animation finishes, the attack is cancelled

The attack animation is cancelled when the dinosaur enters the water, and the attack does not hit or comple

The attack should continue and complete normally when entering the water during the attack animation