diff --git a/analyses/org.eclipse.tracecompass.incubator.tmf.ui.multiview.ui/src/org/eclipse/tracecompass/incubator/internal/tmf/ui/multiview/ui/view/xychart/TreeViewer.java b/analyses/org.eclipse.tracecompass.incubator.tmf.ui.multiview.ui/src/org/eclipse/tracecompass/incubator/internal/tmf/ui/multiview/ui/view/xychart/TreeViewer.java
index ba7d8a9..48afe4c 100644
--- a/analyses/org.eclipse.tracecompass.incubator.tmf.ui.multiview.ui/src/org/eclipse/tracecompass/incubator/internal/tmf/ui/multiview/ui/view/xychart/TreeViewer.java
+++ b/analyses/org.eclipse.tracecompass.incubator.tmf.ui.multiview.ui/src/org/eclipse/tracecompass/incubator/internal/tmf/ui/multiview/ui/view/xychart/TreeViewer.java
@@ -77,7 +77,10 @@
                     && isChecked(element)) {
                 Optional<ITmfTreeDataModel> model = tryGetModel(element);
                 if (model.isPresent()) {
-                    return getLegendImage(model.get().getName());
+                    ITmfTreeDataModel treeModel = model.get();
+                    if (treeModel.hasRowModel()) {
+                        return getLegendImage(treeModel.getId());
+                    }
                 }
             }
             return null;