Make FacetGrid legend more robust