How come they can still fly when they're stunned? I suggest that they'll lose flight ability when they're stunned.
The idea is flight is natural function, not unlike breathing, and one does not stop breathing simply because one is dazed. Such is the pox logic behind magical things still flying.
Well, that logic doesn't work. But logic comes second after balance anyway. If there's no balance reason to change it, don't.