1 files changed, 2 insertions, 2 deletions

diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/NestedPackages.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/NestedPackages.java
index c4541fc6..b0686654 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/NestedPackages.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/NestedPackages.java
@@ -88,10 +88,10 @@ public class NestedPackages {
                 ClassSelectorClassNode node = classToNode.remove(entry);
 
                 if (node != null) {
+                        DefaultMutableTreeNode packageNode = (DefaultMutableTreeNode) node.getParent();
                         node.removeFromParent();
-                        // remove dangling packages
-                        DefaultMutableTreeNode packageNode = packageToNode.get(entry.getPackageName());
 
+                        // remove dangling packages
                         while (packageNode != null && packageNode.getChildCount() == 0) {
                                 DefaultMutableTreeNode theNode = packageNode;
                                 packageNode = (DefaultMutableTreeNode) packageNode.getParent();