Add types to expression nodes

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
index 861e847..8989ef3 100644
--- a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
@@ -1,6 +1,7 @@
 package org.zwobble.couscous.ast;
 
 import org.zwobble.couscous.ast.visitors.ExpressionNodeVisitor;
+import org.zwobble.couscous.values.TypeReference;
 
 import static org.zwobble.couscous.ast.VariableReferenceNode.reference;
 
@@ -19,4 +20,9 @@ public class AssignmentNode implements ExpressionNode {
     public <T> T accept(ExpressionNodeVisitor<T> visitor) {
         return visitor.visit(this);
     }
+
+    @Override
+    public TypeReference getType() {
+        return value.getType();
+    }
 }