Adds back in the country array to make the tests pass