Keep your wits about you

Meaning: To remain alert and think clearly, especially in a difficult or dangerous situation.

Example: Keep your wits about you when walking through an unfamiliar area.