Advice needed

So I EBF my 8 month old and lately it’s bee. SO hard to feed him because he gets extremely distracted (yes I’ve tried everything - dark room, no distractions, etc). He only nurses well before bedtime and nap time. He has enough diapers and is in the 90th percentile for weight, but I’m wondering if I should just pump at this point so I can stop stressing? Also is anyone else in the same boat and have any advice/tips?