diff --git a/bin/shrimp b/bin/shrimp
index 0754565..213e4cb 100755
--- a/bin/shrimp
+++ b/bin/shrimp
@@ -143,7 +143,7 @@ async function main() {
       process.exit(1)
     }
     const input = readFileSync(file, 'utf-8')
-    console.log(treeToString2(parseCode(input), input))
+    console.log(treeToString2(parseCode(input).topNode, input))
     return
   }