Diagnosing electrical lighting issue

Hey everyone,

I am having an issue when I turn off my B6 S4 the car will buzz as if I forgot to turn off my headlights regardless if the the headlight switch is in Auto or Off. When I exit the vehicle the headlights and taillights remain on, and won’t go out. If I turn the car on and drive it a few feet and turn it off again, the problem sometimes clears and the lights turn off normal.

I am leaning towards the headlight switch being on its way out, but I thought I would ask here in case there is another opinion or another common issue I am unaware of. Is there a way to test the switch?

Cheers