1 files changed, 4 insertions, 2 deletions

diff --git a/src/switch.zig b/src/switch.zig
index 058067d..e5d243f 100644
--- a/src/switch.zig
+++ b/src/switch.zig
@@ -2,7 +2,7 @@ const std = @import("std");
 const xdg = @import("xdg");
 const zup = @import("zup");
 
-const Allocator = std.mem.Allocator;
+const Config = zup.Config;
 const Installation = zup.Installation;
 
 pub const params = "<NAME>";
@@ -10,7 +10,9 @@ pub const description = "Switches to a Zig version.  Run `zup list -i` to see in
 pub const min_args = 1;
 pub const max_args = 1;
 
-pub fn main(comptime Result: type, allocator: Allocator, res: Result) !void {
+pub fn main(comptime Result: type, config: Config, res: Result) !void {
+    const allocator = config.allocator;
+    
     const name = res.positionals[0];
     if (!try Installation.isInstalled(allocator, name)) {
         std.log.err(