There is a couple things you could do:
First of all(and most important): Improve your game. If you keep improving your game you will eventually improve your "Unconscious Competence" which basically means you won't even think about those punts.
Second: Improve your mental game, read books like The Mental Game of Poker, Tilt Amour and so on
Third: Try improving your focus! If you are studying for instance.. put away ANYTHING that might distract you. Do not look at your phone, don't surf on YouTube.. don't even look around in your room. Try just focusing on what you do. Go for 30 minutes.. next time for more.. and them some more and more.. your focus will improve and you won't be easly distracted and therefore play your best game.
If you are playing.. play less tables. Pick a player or two and watch every hand they are playing (to keep your mind occupied)!
When breaks come,
take them! Get some fresh air, move around!