diff --git a/generator/main.zig b/generator/main.zig
index c1cc52b..18f7213 100644
--- a/generator/main.zig
+++ b/generator/main.zig
@@ -62,7 +62,7 @@ pub fn main() !void {
     try out_buffer.append(0);
 
     const src = out_buffer.items[0 .. out_buffer.items.len - 1 :0];
-    const tree = try std.zig.parse(allocator, src);
+    const tree = try std.zig.Ast.parse(allocator, src, .zig);
     const formatted = try tree.render(allocator);
     defer allocator.free(formatted);
 
diff --git a/generator/vulkan/build_integration.zig b/generator/vulkan/build_integration.zig
index a8a1eb6..7af8aca 100644
--- a/generator/vulkan/build_integration.zig
+++ b/generator/vulkan/build_integration.zig
@@ -96,7 +96,7 @@ pub const GenerateStep = struct {
         try out_buffer.append(0);
 
         const src = out_buffer.items[0 .. out_buffer.items.len - 1 :0];
-        const tree = try std.zig.parse(self.builder.allocator, src);
+        const tree = try std.zig.Ast.parse(self.builder.allocator, src, .zig);
         std.debug.assert(tree.errors.len == 0); // If this triggers, vulkan-zig produced invalid code.
 
         var formatted = try tree.render(self.builder.allocator);