1 files changed, 7 insertions, 1 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/node/ClassSelectorClassNode.java b/src/main/java/cuchaz/enigma/gui/node/ClassSelectorClassNode.java
index bf6b178..922f8f2 100644
--- a/src/main/java/cuchaz/enigma/gui/node/ClassSelectorClassNode.java
+++ b/src/main/java/cuchaz/enigma/gui/node/ClassSelectorClassNode.java
@@ -17,13 +17,19 @@ import javax.swing.tree.DefaultMutableTreeNode;
 
 public class ClassSelectorClassNode extends DefaultMutableTreeNode {
 
+        private final ClassEntry obfEntry;
         private ClassEntry classEntry;
 
-        public ClassSelectorClassNode(ClassEntry classEntry) {
+        public ClassSelectorClassNode(ClassEntry obfEntry, ClassEntry classEntry) {
+                this.obfEntry = obfEntry;
                 this.classEntry = classEntry;
                 this.setUserObject(classEntry);
         }
 
+        public ClassEntry getObfEntry() {
+                return obfEntry;
+        }
+
         public ClassEntry getClassEntry() {
                 return this.classEntry;
         }