Fix  city count test