Fixed logic to limit counties