1 files changed, 4 insertions, 5 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/MethodImplementationsTreeNode.java b/src/main/java/cuchaz/enigma/analysis/MethodImplementationsTreeNode.java
index bacb1aa..6791b83 100644
--- a/src/main/java/cuchaz/enigma/analysis/MethodImplementationsTreeNode.java
+++ b/src/main/java/cuchaz/enigma/analysis/MethodImplementationsTreeNode.java
@@ -54,11 +54,11 @@ public class MethodImplementationsTreeNode extends DefaultMutableTreeNode {
         }
 
         public String getDeobfClassName() {
-                return this.deobfuscatingTranslator.translateClass(this.entry.getClassName());
+                return this.deobfuscatingTranslator.getTranslatedClass(this.entry.getOwnerClassEntry()).getClassName();
         }
 
         public String getDeobfMethodName() {
-                return this.deobfuscatingTranslator.translate(this.entry);
+                return this.deobfuscatingTranslator.getTranslatedMethod(this.entry).getName();
         }
 
         @Override
@@ -80,9 +80,8 @@ public class MethodImplementationsTreeNode extends DefaultMutableTreeNode {
                 // get all method implementations
                 List<MethodImplementationsTreeNode> nodes = Lists.newArrayList();
                 for (String implementingClassName : index.getImplementingClasses(this.entry.getClassName())) {
-                        MethodEntry methodEntry = new MethodEntry(new ClassEntry(implementingClassName), this.entry.getName(), this.entry.getSignature()
-                        );
-                        if (index.containsObfBehavior(methodEntry)) {
+                        MethodEntry methodEntry = new MethodEntry(new ClassEntry(implementingClassName), this.entry.getName(), this.entry.getDesc());
+                        if (index.containsObfMethod(methodEntry)) {
                                 nodes.add(new MethodImplementationsTreeNode(this.deobfuscatingTranslator, methodEntry));
                         }
                 }