keep (all) your wits about you
US /kiːp ɔl jʊər wɪts əˈbaʊt ju/
UK /kiːp ɔl jʊər wɪts əˈbaʊt ju/

1.
to stay alert and be able to think quickly in a difficult or dangerous situation
:
•
You need to keep all your wits about you when driving in heavy traffic.
•
It's a dangerous situation, so keep your wits about you.