Glad you're feeling a little better. When I have those situations It's important for me to not let my mind wander and be in control, rather, I do my best to only allow my mind to go where I want it to.
Yes, that's easier said than done. But, with practice it can certainly help.