Only use moveClassTree for ClassEntry (Fix #10)

1 files changed, 5 insertions, 3 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/GuiController.java b/src/main/java/cuchaz/enigma/gui/GuiController.java
index d6c170bd..1deb7d2a 100644
--- a/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -198,7 +198,7 @@ public class GuiController {
         this.deobfuscator.rename(obfReference.getNameableEntry(), newName);
         this.isDirty = true;
 
-        if (refreshClassTree)
+        if (refreshClassTree && deobfReference.entry instanceof ClassEntry)
             this.gui.moveClassTree(deobfReference, newName);
         refreshCurrentClass(obfReference);
 
@@ -208,7 +208,8 @@ public class GuiController {
         EntryReference<Entry, Entry> obfReference = this.deobfuscator.obfuscateReference(deobfReference);
         this.deobfuscator.removeMapping(obfReference.getNameableEntry());
         this.isDirty = true;
-        this.gui.moveClassTree(deobfReference, obfReference.entry.getName(), false, true);
+        if (deobfReference.entry instanceof ClassEntry)
+            this.gui.moveClassTree(deobfReference, obfReference.entry.getName(), false, true);
         refreshCurrentClass(obfReference);
     }
 
@@ -216,7 +217,8 @@ public class GuiController {
         EntryReference<Entry, Entry> obfReference = this.deobfuscator.obfuscateReference(deobfReference);
         this.deobfuscator.markAsDeobfuscated(obfReference.getNameableEntry());
         this.isDirty = true;
-        this.gui.moveClassTree(deobfReference, obfReference.entry.getName(), true, false);
+        if (deobfReference.entry instanceof ClassEntry)
+            this.gui.moveClassTree(deobfReference, obfReference.entry.getName(), true, false);
         refreshCurrentClass(obfReference);
     }