1 files changed, 2 insertions, 2 deletions

diff --git a/src/cuchaz/enigma/gui/ClassSelectorClassNode.java b/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
index f054188..6c1c9a0 100644
--- a/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
+++ b/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
@@ -30,8 +30,8 @@ public class ClassSelectorClassNode extends DefaultMutableTreeNode {
         
         @Override
         public String toString() {
-                if (m_classEntry instanceof DecoratedClassEntry) {
-                        return ((DecoratedClassEntry)m_classEntry).getDecoration() + m_classEntry.getSimpleName();
+                if (m_classEntry instanceof ScoredClassEntry) {
+                        return String.format("%d%% %s", (int)((ScoredClassEntry)m_classEntry).getScore(), m_classEntry.getSimpleName());
                 }
                 return m_classEntry.getSimpleName();
         }