Dying with keepInventory on while using an Elytra or flying with a riptide enchanted trident

If you die with the gamerule keepInventory set to true while flying with an elytra or a riptide enchanted trident, you will respawn in the elytra flying state/riptide spinning state but on the ground. This happened to me playing on a realm. To fix it I had to reconnect to the server.