Looking up while flying with Elytra in water causes world to break and disconnect

I was testing the new animations in the latest snapshot and realized that when you fly into water with elytra, the animation is a bit buggy. But when you look up while flying with it in the water, you instantly sink down below the blocks, and the world disconnects and crashes.