1 files changed, 4 insertions, 2 deletions

diff --git a/src/remove.zig b/src/remove.zig
index ad1a610..f795df7 100644
--- a/src/remove.zig
+++ b/src/remove.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 =
@@ -13,7 +13,9 @@ pub const description = "Removes an installed Zig version.  Run `zup list -i` to
 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)) {