Swirling Animation doesn't stop when player die

Player remain in low and animation of riptide didn't stop

Steps to reproduce:
1) Get a trident with Riptide enchant

2) Make it rain

3) Charge using the Riptide twice in the sky

4) Charge when you're about to fall to trigger animation

5) Ensure you die while in animation