Maybe give the lights an ID and use that as a switch as well.