3 files changed, 7 insertions, 1 deletions
diff --git a/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/IRGenerator.java b/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/IRGenerator.java
index f8a5756..657a817 100644
--- a/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/IRGenerator.java
+++ b/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/IRGenerator.java
@@ -29,7 +29,7 @@ public class IRGenerator {
   }
 
   public List<Instruction> generate() {
-    Program program = (Program) ast.getRoot();
+    Program program = (Program) ast.root();
 
     for (Statement stmt : program.statements) {
       generateStatement(stmt);
diff --git a/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/Instructions/Alloc.java b/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/Instructions/Alloc.java
index cda3a56..fd6e7b7 100644
--- a/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/Instructions/Alloc.java
+++ b/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/Instructions/Alloc.java
@@ -1,10 +1,13 @@
 package no.eliashaugsbakk.kompilator.IRGeneration.Instructions;
 
+import org.jetbrains.annotations.NotNull;
+
 // allocate a new variable: set x: string = "Hello"; mut y: string?;
 //
 // Alloc("x", "string", false, "Hello") // Alloc("y", "string", true, "null")
 public record Alloc(String name, String type, boolean mutable, String initializer) implements Instruction {
   @Override
+  @NotNull
   public String toString() {
     return "Alloc[name=" + name + ", type=" + type + ", mutable=" + mutable + ", initializer=\"" +
         initializer + "\"]";
diff --git a/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/Instructions/Assign.java b/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/Instructions/Assign.java
index 80a6d9e..eb00604 100644
--- a/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/Instructions/Assign.java
+++ b/src/main/java/no/eliashaugsbakk/kompilator/IRGeneration/Instructions/Assign.java
@@ -1,9 +1,12 @@
 package no.eliashaugsbakk.kompilator.IRGeneration.Instructions;
 
+import org.jetbrains.annotations.NotNull;
+
 // assign a value to a variable: x = "Hello";
 // Assign("x", "Hello")
 public record Assign(String name, String value) implements Instruction {
   @Override
+  @NotNull
   public String toString() {
     return "Assign[name=" + name + ", value=\"" + value + "\"]";
   }