Fix logic bug in previous checkin. Add array to set