Brake pedal won't return

My brake lights occasionally stay on, an ever so slight lift of the pedal shuts them off. But its a nuisance. I've tried adjusting the switch and it's due to the small button on the switch not sitting/seating properly all the time on the pedal. I may look at putting a spring on it.