Allow descendants of nodes to be traversed

1 files changed, 4 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNode.java b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
index d2c64f5..c252bb3 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
@@ -1,8 +1,8 @@
 package org.zwobble.couscous.ast;
 
-import java.util.List;
+import org.zwobble.couscous.ast.visitors.NodeMapper;
 
-import org.zwobble.couscous.ast.visitors.NodeVisitor;
+import java.util.List;
 
 public class ClassNode implements Node {
     public static ClassNodeBuilder builder(String name) {
@@ -54,8 +54,8 @@ public class ClassNode implements Node {
     }
 
     @Override
-    public void accept(NodeVisitor visitor) {
-        visitor.visit(this);
+    public <T> T accept(NodeMapper<T> visitor) {
+        return visitor.visit(this);
     }
 
     @Override