diff --git a/more_math/ParseTree.py b/more_math/ParseTree.py
index 7f026a5..7523074 100644
--- a/more_math/ParseTree.py
+++ b/more_math/ParseTree.py
@@ -1,9 +1,9 @@
 import antlr4
 from comfy_api.latest import io
 
-@io.comfytype(io_type="PARSER")
+@io.comfytype(io_type="SYNTAX_TREE")
 class MrmthParseTree(io.ComfyTypeIO):
-    Type = antlr4.Parser  # Python type hint
+    Type = antlr4.ParseTreeVisitor  # Python type hint
 
     class Input(io.Input):
         def __init__(self, id: str, **kwargs):