Distracted eater

Mi

It is so hard to get my son to just focus on nursing these days! He’s constantly unlatching to look all around. I’m worried my supply will drop because he went from nursing every hour for a solid 15 minutes to nursing every 2-3 hours and only for 12ish minutes with frequent unlatching and not fully draining my breast. Is anyone else experiencing something similar? Did it effect your supply? Does anyone have any tips on how to correct this behavior? He’s 3 1/2 months EBF. Should I pump after feeding? I’ll be returning to work soon so I’m extremely worried about my supply diminishing.

TIA.