Keep one's wits about one

(See: Keep one's eyes peeled)

Meanings
Fig.: to pull oneself together for action