diff --git a/lib.nix b/lib.nix
index e7e0648..b9a0406 100644
--- a/lib.nix
+++ b/lib.nix
@@ -39,14 +39,20 @@ let
 
   getSiteGatewayV4 = siteName:
     let site = entities.sites."${siteName}";
-    in if hasAttr "local-gateway" site then getHostIpv4 else site.gateway-v4;
+    in if hasAttr "local-gateway" site then
+      getHostIpv4 site.local-gateway
+    else
+      site.gateway-v4;
 
   getHostGatewayV4 = hostname:
     getSiteGatewayV4 entities.hosts."${hostname}".site;
 
   getSiteGatewayV6 = siteName:
     let site = entities.sites."${siteName}";
-    in if hasAttr "local-gateway" site then getHostIpv6 else site.gateway-v6;
+    in if hasAttr "local-gateway" site then
+      getHostIpv6 site.local-gateway
+    else
+      site.gateway-v6;
 
   getHostGatewayV6 = hostname:
     getSiteGatewayV6 entities.hosts."${hostname}".site;