Allow descendants of nodes to be traversed

1 files changed, 4 insertions, 3 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ConstructorNode.java b/src/main/java/org/zwobble/couscous/ast/ConstructorNode.java
index c94554b..2e8b85a 100644
--- a/src/main/java/org/zwobble/couscous/ast/ConstructorNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ConstructorNode.java
@@ -2,6 +2,7 @@ package org.zwobble.couscous.ast;
 
 import java.util.List;
 
+import org.zwobble.couscous.ast.visitors.NodeMapper;
 import org.zwobble.couscous.ast.visitors.NodeVisitor;
 
 public class ConstructorNode implements CallableNode {
@@ -28,10 +29,10 @@ public class ConstructorNode implements CallableNode {
     public List<StatementNode> getBody() {
         return body;
     }
-    
+
     @Override
-    public void accept(NodeVisitor visitor) {
-        visitor.visit(this);
+    public <T> T accept(NodeMapper<T> visitor) {
+        return visitor.visit(this);
     }
 
     @Override