Kai's animation is broken

When you use Kai's "Lunge" ability in the dinosaur packs, the animation plays like he's on land, unlike other dinosaurs, whose animation works properly in the water. After the animation ends, his head keeps bobbing up and down.

Kai's animation played like he was on land, unlike other aquatic dinosaurs, and after the animation ended, his head kept moving up and down.

Kai's "Lunge" animation should've played like it does underwater for other aquatic dinosaurs, without his head twitching up and down.