diff --git a/src/main.zig b/src/main.zig
index 052c082..3ac5f71 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -6,6 +6,6 @@ const hosts = @import("hosts.zig");
 pub fn main() !void {
     try util.check_perms();
     const domain, const ip_ver = try util.get_input();
-    const ips = try mdns.get_mdns(domain);
-    try hosts.update_hosts(ips, ip_ver);
+    const ips = try mdns.get_mdns(domain, ip_ver);
+    try hosts.update_hosts(ips);
 }