I've created a scoreboard objective that link to the stat.useItem.minecraft.fireworks tracker. It works well till i use it mid-air while flying on elytra. It seems that developers missed that detail when they added it to the game.