diff --git a/tree/.local/nu/mullvad.nu b/tree/.local/nu/mullvad.nu
index 03cf82f..f3d1386 100644
--- a/tree/.local/nu/mullvad.nu
+++ b/tree/.local/nu/mullvad.nu
@@ -53,17 +53,10 @@ def dns [
     # HACK: turn the optional argument into a list. not sure how to pass it to nslookup otherwise.
     let server = ([$server] | compact)
 
-    nslookup $name ...$server |
-      lines |
-      skip until { |line| $line =~ answer } |
-      parse -r "Address:\\s(?<ipv4>\\d+\\.\\d+\\.\\d+\\.\\d+)|Address:\\s(?<ipv6>[\\d:a-z]+)" |
-      each { |row|
-        if $row.ipv4 != "" {
-          { kind: ipv4, addr: $row.ipv4 }
-        } else {
-          { kind: ipv6, addr: $row.ipv6 }
-        }
-      }
+    let ipv4 = (dig $name A    | lines | skip until {|l| $l =~ "ANSWER SECTION" } | skip 1 | take until { |l| $l == "" } | split column "\t" -c | rename host ttl _ type addr) | reject _
+    let ipv6 = (dig $name AAAA | lines | skip until {|l| $l =~ "ANSWER SECTION" } | skip 1 | take until { |l| $l == "" } | split column "\t" -c | rename host ttl _ type addr) | reject _
+
+    $ipv4 | append $ipv6
 }
 
 
@@ -86,7 +79,7 @@ def "m daemon" [
         $cargo_flags ++= [--features api-override]
         $envs ++= [
             MULLVAD_API_HOST=api.stagemole.eu
-            MULLVAD_API_ADDR=(dns api.stagemole.eu | where kind == ipv4 | get 0.addr):443
+            MULLVAD_API_ADDR=(dns api.stagemole.eu | where type == A | get 0.addr):443
         ]
     }