diff --git a/zml/aio.zig b/zml/aio.zig
index 82d1858..56fc14e 100644
--- a/zml/aio.zig
+++ b/zml/aio.zig
@@ -518,7 +518,7 @@ pub fn loadBuffers(
 /// The `init_args` are used to initialize the non Buffer fields, using `Model.init` function.
 pub fn loadBuffersWithPrefix(
     comptime Model: type,
-    init_args: anytype,
+    init_args: if (@hasDecl(Model, "init")) stdx.meta.Tail(stdx.meta.FnArgs(Model.init)) else void,
     buffer_store: BufferStore,
     allocator: std.mem.Allocator,
     platform: zml.Platform,