Fix test to address city column