If you throw a Riptide-enchanted trident and log out, the riptide effect is cut short

The bug

If you have a trident with the Riptide enchantment, and you throw it to fly with it, the "flight" will be cut short if you log out and log back in.

To reproduce

1. Obtain a trident enchanted with the Riptide enchantment
2. Stand in water
3. Throw the trident
4. Exit the world
5. Re-enter the world

Observed behavior

The velocity of the player is not anymore as it was before the world was exited.

Expected behavior

The velocity of the player would match what it was before the world was exited.