diff --git a/src/main/java/ui/tree/JDiagramTree.java b/src/main/java/ui/tree/JDiagramTree.java
index 14b9f463060ce0dc1687b9409b3367e6808a8aca..6e64906f453d1bbf0ad8fde600f94fab6faeb871 100755
--- a/src/main/java/ui/tree/JDiagramTree.java
+++ b/src/main/java/ui/tree/JDiagramTree.java
@@ -261,8 +261,9 @@ public class JDiagramTree extends javax.swing.JTree implements ActionListener, M
                 if (selectedST.getType() == SimulationTrace.XML_DIPLO) {
                 	jmiCompareST = new JMenuItem("Compare");
                 	jmiCompareST.addActionListener(this);
+                    popupTreeST.add(jmiCompareST);
                 }
-                popupTreeST.add(jmiCompareST);
+
 
 
             //}