Fix TrafficLight#state_at