diff --git a/src/main/java/org/JesusFreke/smali/smali.java b/src/main/java/org/JesusFreke/smali/smali.java
index e9b47476..2fa5a563 100644
--- a/src/main/java/org/JesusFreke/smali/smali.java
+++ b/src/main/java/org/JesusFreke/smali/smali.java
@@ -179,6 +179,7 @@ public class smali
         CommonTree t = (CommonTree) result.getTree();
 
         CommonTreeNodeStream treeStream = new CommonTreeNodeStream(t);
+        treeStream.setTokenStream(tokens);
 
         smaliTreeWalker dexGen = new smaliTreeWalker(treeStream);