1 files changed, 5 insertions, 3 deletions

diff --git a/src/update.zig b/src/update.zig
index a8fd075..3451d90 100644
--- a/src/update.zig
+++ b/src/update.zig
@@ -1,7 +1,7 @@
 const std = @import("std");
 const zup = @import("zup");
 
-const Allocator = std.mem.Allocator;
+const Config = zup.Config;
 const Installation = zup.Installation;
 
 // TODO: A way to specify a subset?
@@ -11,11 +11,13 @@ pub const description = "Updates all installed Zig versions.";
 pub const min_args = 0;
 pub const max_args = 0;
 
-pub fn main(comptime Result: type, allocator: Allocator, _: Result) !void {
+pub fn main(comptime Result: type, config: Config, _: Result) !void {
+    const allocator = config.allocator;
+    
     var installed = try Installation.getInstalledList(allocator);
     defer Installation.deinitMap(allocator, &installed);
 
-    var available = try Installation.getAvailableList(allocator);
+    var available = try Installation.getAvailableList(config);
     defer Installation.deinitMap(allocator, &available);
 
     var it = installed.iterator();