Firework boost durations doesnt get reset when the player hits the ground while the duration timer is going

I was practising the spear then I hit the ground, when I started flying again the firework boost kept my momentum