1 files changed, 8 insertions, 0 deletions
diff --git a/src/main/java/org/zwobble/couscous/ast/IfStatementNode.java b/src/main/java/org/zwobble/couscous/ast/IfStatementNode.java
index 8b4709a..972f0a4 100644
--- a/src/main/java/org/zwobble/couscous/ast/IfStatementNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/IfStatementNode.java
@@ -1,5 +1,6 @@
 package org.zwobble.couscous.ast;
 
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
 import org.zwobble.couscous.ast.visitors.StatementNodeMapper;
 
 import java.util.List;
@@ -53,6 +54,13 @@ public class IfStatementNode implements StatementNode {
             eagerMap(falseBranch, statement -> statement.replaceExpressions(replace)));
     }
 
+    public StatementNode transform(NodeTransformer transformer) {
+        return new IfStatementNode(
+            transformer.visit(condition),
+            transformer.transformStatements(trueBranch),
+            transformer.transformStatements(falseBranch));
+    }
+
     @Override
     public String toString() {
         return "IfStatementNode(condition=" + condition + ", trueBranch="