Firework boost remains after landing

I used a firework rocket with a flight duration of 127 to demonstrate the bug, but it also occurs in survival mode without any cheats and with a flight duration of 3 (or less).

What I expected to happen:
The boost of a firework rocket vanishes after landing

What actually happened:
After landing, particles still come out of my body. When I launch myself up again, I am immediately boosted

How to reproduce

1. Equip an elytra and give yourself a long duration rocket.

   /give @s minecraft:firework_rocket{Fireworks:{Flight:127b}}
   

2. Active the elytra and shoot a firework rocket.
3. Land on the ground and active the elytra again.
   → You are being boosted again

Code analysis

Code analysis by Oval can be found in this comment