Nipple shield

I have a question ladies. My baby had a tongue tie while in the NICU, so I was told to use a nipple shield as well as pump for 20 minutes after each feeding to fully empty my breasts. My baby would nurse with the shield for 20 mins and still take a bottle afterwards. But now my baby has been latching better and uses the shield half the time instead of constantly, getting full and not needing to be supplemented with a bottle, now only bottle feeding at night to give me a break. When I pump after each feed I am still getting about 6oz every time. My question is now that my baby is nursing better, do I still need to pump so often ? I’m afraid of getting mastitis again which I developed at 1 week pp. but I’m sick of pumping every time I nurse. Sorry for the long post hopefully someone reads this 😬