Legend diagrams fix :sparkles: