better fix for #2

1 files changed, 3 insertions, 4 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/JarIndex.java b/src/main/java/cuchaz/enigma/analysis/JarIndex.java
index 96f92c3d..51a2543a 100644
--- a/src/main/java/cuchaz/enigma/analysis/JarIndex.java
+++ b/src/main/java/cuchaz/enigma/analysis/JarIndex.java
@@ -704,18 +704,17 @@ public class JarIndex {
     }
 
     public Set<String> getImplementingClasses(String targetInterfaceName) {
-        
+
         // linear search is fast enough for now
         Set<String> classNames = Sets.newHashSet();
-        Set<String> interfaceNames = Sets.newHashSet(targetInterfaceName);
         for (Map.Entry<ClassEntry, ClassEntry> entry : this.translationIndex.getClassInterfaces()) {
             ClassEntry classEntry = entry.getKey();
             ClassEntry interfaceEntry = entry.getValue();
-            if (interfaceNames.contains(interfaceEntry.getName())) {
+            if (interfaceEntry.getName().equals(targetInterfaceName)) {
                 String className = classEntry.getClassName();
                 classNames.add(className);
                 if (isInterface(className)) {
-                    interfaceNames.add(className);
+                    classNames.addAll(getImplementingClasses(className));
                 }
 
                 this.translationIndex.getSubclassNamesRecursively(classNames, classEntry);