1 files changed, 5 insertions, 2 deletions

diff --git a/src/install.zig b/src/install.zig
index ebf1c01..f9c7309 100644
--- a/src/install.zig
+++ b/src/install.zig
@@ -5,6 +5,7 @@ const xdg = @import("xdg");
 const zup = @import("zup");
 
 const Allocator = std.mem.Allocator;
+const Config = zup.Config;
 const ArchiveRead = libarchive.Read;
 const Installation = zup.Installation;
 const Installations = zup.Installations;
@@ -17,8 +18,10 @@ pub const description = "Installs a Zig version.  Run `zup list -i` to see insta
 pub const min_args = 1;
 pub const max_args = 1;
 
-pub fn main(comptime Result: type, allocator: Allocator, res: Result) !void {
-    var available = try Installation.getAvailableList(allocator);
+pub fn main(comptime Result: type, config: Config, res: Result) !void {
+    const allocator = config.allocator;
+    
+    var available = try Installation.getAvailableList(config);
     defer Installation.deinitMap(allocator, &available);
 
     return perform(allocator, res.positionals[0], res.args.force, available);