Elytra animation continues when gliding into water but movement doesn't

Elytra animation continues when gliding into water but movement doesn't.

What I expected to happen was for the animation to stop and move to the normal swimming animation.

What actually happend was the animation stayed as the gliding with the elytra animation even though the movement didn't. Also when leaving the water the elytra movement happends until you touch the ground.

Steps to Reproduce:
1. start elytra gliding from any height (as long as it's in the animation)
2. whilst gliding land in water
3. see in third person perspective that the animation for gliding is happening instead of the animation for swimming.
4. leave water and glide for a block.

The animation resets back to walking when you get out the water and onto land thus making this issue not fatal to the game.

Video was to large to upload so uploaded it to youtube (video will be live 63 minutes after submission) : https://youtu.be/Xilx6CXGMh0