I was flying with Elytra and when i Dove underwater and then back up, i fell straight down.

1. Equip Elytra
2. jump off of a cliff
3. dive under water
4. fly out out of water straight up