1 files changed, 3 insertions, 1 deletions

diff --git a/src/ir.c b/src/ir.c
index 86a1a9a..7f841be 100644
--- a/src/ir.c
+++ b/src/ir.c
@@ -156,7 +156,9 @@ generate_basm(ParseTree *parse_tree) {
     array_init(program->lines, 0);
     for (size_t i = 0; i < array_size(parse_tree->roots); ++i) {
         Node *root = parse_tree->roots[i];
-        emit_basm(program, root);
+        Operand ret = emit_basm(program, root);
+        LineInfo line = (LineInfo){.line = root->line, .col = root->col};
+        EMIT_0(program, line, OP_RETURN, ret);
     }
     return program;
 }