overnight feeding/supply question

Ke
LO has started sleeping thru the night at 10w (yay!) but my milk supply is not on the same page! I've been waking up to pump once in the middle of the night otherwise I'll wake up with a soaked shirt (I do wear nursing pads too.)
Question is, should I try to stop waking to pump/relieve... or what? 
I go back to work in two weeks so do want to have as much supply as possible so I'm fine waking up once in the middle of the night for that reason...
Advice? Thoughts?