The stock pot is inexpensive and actually sounds fine, but tracking, especially at low volume, can be a problem. I have several of these pots and they measure all over the place. It wouldnt be a surprise for a bad one to pop up here and there. That said, for a very affordable replacement that tracks well even at low volume, the Alps "blue velvet" pots are good. They have pins for thru-hole mount which makes them a little harder to solder wire to them, but very doable. Of course, there are more expensive pots that are more highly regarded and plenty of debate about the "sound" of pots. But that aside, for about $12 (Mouser) the Alps pot should solve your tracking / channel inbalance problems. They are pretty smooth, track well at low volume and all the way through the range and sound fine. Personally, I dont like the idea of a stepped control on a headphone amp unless it is something with a lot of steps (40+) which is expensive. I think having fine control over the volume is necessary in a headphone amp. But just my opinion.
Alps RK27 series pot ("Blue Velvet") , 100K "audio taper" ... http://www.mouser.com/ProductDetail/ALPS/RK27112A00AK/?qs=sGAEpiMZZMuarXojuOTscwVDmX9fHUSLvlV84oHMUXs%3d
BTW, the shaft diameter is 6mm, not 9mm as listed on the Mouser site. So the control knob does fit just fine.