Do you punish an unbalanced party?

[Pleh]

Never forget terrain advantages.

Always leave the party a backup plan in case things go awry. I mean, it's a fantastic and common trope in fantasy that the enemy swoops down in a flying ambush, so the heroes fight for a short time while looking for cover. If they can escape to a small, enclosed space, the enemy will have to abandon their advantage to be able to continue the pursuit.

Not every encounter has to be defeated by killing the monsters. Sometimes just surviving the ordeal is worth the same experience.

It can be difficult outrunning a flying creature, so pick poor or clumsy maneuverability for this trope. That gives the players a moment to take action while the enemy flies off to make a wide turn to come around again.