If you use a firework rocket of high flight duration (i used flight duration 127), then exit flight mode, your arm will still have rocket particles, and when you re-enter flight mode, you will still have the flight duration boost. If you don’t re-enter flight mode after using the rocket, the duration will still continue to be used up even when not in flight mode.