1 files changed, 6 insertions, 3 deletions
diff --git a/boot/validation.cc b/boot/validation.cc
index b046b4e..91dc064 100644
--- a/boot/validation.cc
+++ b/boot/validation.cc
@@ -79,9 +79,12 @@ namespace elna::boot
         {
             procedure->accept(this);
         }
-        for (auto *entry_statement : unit->entry_point)
+        if (unit->entry_point.has_value())
         {
-            entry_statement->accept(this);
+            for (auto *entry_statement : unit->entry_point->statements)
+            {
+                entry_statement->accept(this);
+            }
         }
     }
 
@@ -91,7 +94,7 @@ namespace elna::boot
         {
             auto procedure = this->bag.lookup(declaration->identifier.name())->is_procedure();
             this->bag.enter(procedure->scope);
-            for (auto *statement : declaration->body.value().entry_point)
+            for (auto *statement : declaration->body.value().statements)
             {
                 statement->accept(this);
             }