An easy fix would be giving a flying character a 100% dodge against melee attacks with a proper animation (pulling back a little) unless restrained. That is "not cool" but at least it is very easy to implement I think.