Update country field and make sure it works