1 files changed, 3 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/gui/ClassSelectorClassNode.java b/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
index 66e931b..b3d7b89 100644
--- a/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
+++ b/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
@@ -30,6 +30,9 @@ public class ClassSelectorClassNode extends DefaultMutableTreeNode {
         
         @Override
         public String toString() {
+                if (m_classEntry instanceof DecoratedClassEntry) {
+                        return ((DecoratedClassEntry)m_classEntry).getDecoration() + m_classEntry.getSimpleName();
+                }
                 return m_classEntry.getSimpleName();
         }
 }