Add types to expression nodes

1 files changed, 7 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/values/UnitValue.java b/src/main/java/org/zwobble/couscous/values/UnitValue.java
index 14bf327..b4f6974 100644
--- a/src/main/java/org/zwobble/couscous/values/UnitValue.java
+++ b/src/main/java/org/zwobble/couscous/values/UnitValue.java
@@ -1,6 +1,8 @@
 package org.zwobble.couscous.values;
 
 public class UnitValue implements PrimitiveValue {
+    public static final TypeReference REF = TypeReference.typeRef("org.zwobble.couscous.runtime.Unit");
+    
     static final UnitValue UNIT = new UnitValue();
     
     private UnitValue() {
@@ -10,4 +12,9 @@ public class UnitValue implements PrimitiveValue {
     public <T> T accept(PrimitiveValueVisitor<T> visitor) {
         return visitor.visit(this);
     }
+
+    @Override
+    public TypeReference getType() {
+        return REF;
+    }
 }