Throw exception if assigning value of wrong type

1 files changed, 4 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/couscous/interpreter/StackFrameBuilder.java b/src/main/java/org/zwobble/couscous/interpreter/StackFrameBuilder.java
index ac425cd..658bf42 100644
--- a/src/main/java/org/zwobble/couscous/interpreter/StackFrameBuilder.java
+++ b/src/main/java/org/zwobble/couscous/interpreter/StackFrameBuilder.java
@@ -10,20 +10,20 @@ import com.google.common.collect.ImmutableMap;
 import java.util.Optional;
 
 public class StackFrameBuilder {
-    private final ImmutableMap.Builder<Integer, Optional<InterpreterValue>> variables
+    private final ImmutableMap.Builder<VariableNode, Optional<InterpreterValue>> variables
         = ImmutableMap.builder();
     
     public StackFrameBuilder declare(VariableNode variable, InterpreterValue value) {
-        variables.put(variable.getId(), Optional.of(value));
+        variables.put(variable, Optional.of(value));
         return this;
     }
     
     public StackFrameBuilder declare(VariableNode variable) {
-        variables.put(variable.getId(), Optional.empty());
+        variables.put(variable, Optional.empty());
         return this;
     }
     
-    public Map<Integer, Optional<InterpreterValue>> build() {
+    public Map<VariableNode, Optional<InterpreterValue>> build() {
         return variables.build();
     }
 }