diff --git a/src/generate.zig b/src/generate.zig
index 3d7d799..a8a5510 100644
--- a/src/generate.zig
+++ b/src/generate.zig
@@ -174,18 +174,12 @@ const Header = struct {
     };
 };
 
-pub fn main(init: std.process.Init.Minimal) !void {
-    // Allocator and command line args
-    var gpa = std.heap.GeneralPurposeAllocator(.{
-        .thread_safe = false,
-    }){};
-    defer std.debug.assert(gpa.deinit() == .ok);
-    const allocator = gpa.allocator();
-
+pub fn main(init: std.process.Init) !void {
+    const allocator = init.gpa;
     var threaded_io: Io.Threaded = .init_single_threaded;
     const io = threaded_io.io();
 
-    var args = try init.args.iterateAllocator(allocator);
+    var args = try init.minimal.args.iterateAllocator(allocator);
     defer args.deinit();
     std.debug.assert(args.skip());
     const in_path = args.next().?;