Add types to expression nodes

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java b/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java
index 0386a32..8b30024 100644
--- a/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java
@@ -1,6 +1,7 @@
 package org.zwobble.couscous.ast;
 
 import org.zwobble.couscous.ast.visitors.ExpressionNodeVisitor;
+import org.zwobble.couscous.values.TypeReference;
 
 import lombok.Value;
 
@@ -14,4 +15,9 @@ public class TernaryConditionalNode implements ExpressionNode {
     public <T> T accept(ExpressionNodeVisitor<T> visitor) {
         return visitor.visit(this);
     }
+
+    @Override
+    public TypeReference getType() {
+        return ifTrue.getType();
+    }
 }