Also patch for instructions

malcolmstill · Jul 23, 2023 · 779d9c4 · 779d9c4
1 parent f70b9c5
commit 779d9c4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/module/parser.zig b/src/module/parser.zig
@@ -68,6 +68,7 @@ pub const Parser = struct {
 
         // Patch last end so that it is return
         self.module.parsed_code.items[self.module.parsed_code.items.len - 1] = .@"return";
+        self.module.instructions.items[self.module.instructions.items.len - 1] = VirtualMachine.@"return";
 
         return Parsed{ .start = code_start, .max_depth = self.validator.max_depth };
     }
@@ -111,6 +112,7 @@ pub const Parser = struct {
 
         // Patch last end so that it is return
         self.module.parsed_code.items[self.module.parsed_code.items.len - 1] = .@"return";
+        self.module.instructions.items[self.module.instructions.items.len - 1] = VirtualMachine.@"return";
 
         return Parsed{ .start = code_start, .max_depth = self.validator.max_depth };
     }