Fix crash on remapping >1 level of nested class (#262)

1 files changed, 5 insertions, 3 deletions

diff --git a/enigma/src/main/java/cuchaz/enigma/bytecode/translators/TranslationSignatureVisitor.java b/enigma/src/main/java/cuchaz/enigma/bytecode/translators/TranslationSignatureVisitor.java
index bc4a2f0..6cab22c 100644
--- a/enigma/src/main/java/cuchaz/enigma/bytecode/translators/TranslationSignatureVisitor.java
+++ b/enigma/src/main/java/cuchaz/enigma/bytecode/translators/TranslationSignatureVisitor.java
@@ -1,11 +1,12 @@
 package cuchaz.enigma.bytecode.translators;
 
-import cuchaz.enigma.Enigma;
-import org.objectweb.asm.signature.SignatureVisitor;
-
 import java.util.Stack;
 import java.util.function.Function;
 
+import org.objectweb.asm.signature.SignatureVisitor;
+
+import cuchaz.enigma.Enigma;
+
 public class TranslationSignatureVisitor extends SignatureVisitor {
         private final Function<String, String> remapper;
 
@@ -31,6 +32,7 @@ public class TranslationSignatureVisitor extends SignatureVisitor {
                 if (!name.startsWith(lastClass+"$")){//todo see if there's a way to base this on whether there were type params or not
                         name = lastClass+"$"+name;
                 }
+                classStack.push(name);
                 String translatedEntry = this.remapper.apply(name);
                 if (translatedEntry.contains("/")){
                         translatedEntry = translatedEntry.substring(translatedEntry.lastIndexOf("/")+1);