Tweak inheritance and implementation tree generation

1 files changed, 2 insertions, 2 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/MethodImplementationsTreeNode.java b/src/main/java/cuchaz/enigma/analysis/MethodImplementationsTreeNode.java
index e4b0304..b09f7ac 100644
--- a/src/main/java/cuchaz/enigma/analysis/MethodImplementationsTreeNode.java
+++ b/src/main/java/cuchaz/enigma/analysis/MethodImplementationsTreeNode.java
@@ -71,8 +71,8 @@ public class MethodImplementationsTreeNode extends DefaultMutableTreeNode {
                 EntryIndex entryIndex = index.getEntryIndex();
                 InheritanceIndex inheritanceIndex = index.getInheritanceIndex();
 
-                Collection<ClassEntry> inheritors = inheritanceIndex.getChildren(entry.getParent());
-                for (ClassEntry inheritor : inheritors) {
+                Collection<ClassEntry> descendants = inheritanceIndex.getDescendants(entry.getParent());
+                for (ClassEntry inheritor : descendants) {
                         MethodEntry methodEntry = entry.withParent(inheritor);
                         if (entryIndex.hasMethod(methodEntry)) {
                                 nodes.add(new MethodImplementationsTreeNode(translator, methodEntry));