Brake light comes on but goes off after car is restarted?

Could be the switch