Flying with fireworks are broken

Sometimes trying to fly with fireworks, doesn't work
How to reproduce: Take a elytra and equip it, later, take a firework and try to fly with they and sometimes fireworks doesn't work properly!