Yes, and remember that the enhancement needs to be net positive for both survival AND reproduction.

There are clear reproductive advantages to distraction. If your actions prioritize your own wellbeing (or the furthering of some cognitive task) over the instincts to eat, reproduce, and protect your children, your genetics will be at a disadvantage.