There's nothing wrong with replacing bearings in a headset, its daft to replace your headset if there isn't anything wrong with it other than something which should be expected every few 1000s miles!
Just make sure you use good grease! (Juice Lubes Bearing Juice is rad)
Cheers man! See you tomorrow.