1 files changed, 3 insertions, 1 deletions
diff --git a/src/codegen.zig b/src/codegen.zig
index 5636dbd..59ef03b 100644
--- a/src/codegen.zig
+++ b/src/codegen.zig
@@ -223,7 +223,9 @@ test "Codegen exit" {
     var pop = symb.Populator.init(std.testing.allocator);
     var treeNode = parseTree.asNode();
     try pop.populateSymtable(&treeNode);
-    var gen = Generator.init(std.testing.allocator, parseTree);
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    var gen = Generator.init(arena.allocator(), parseTree);
     defer gen.deinit();
     const actual = try gen.generate();
     try expect(std.mem.eql(u8, actual, expected));