Player model is laggy when using firework in elytra-mode in water (F5)

Using firework in water when in elytra mode make your 3d models arms and legs spin a lot. Try it (screenshot not really showing the issue, but couldn't do better)

EDIT: just swimming works