Fix geo_id KeyError