diff --git a/lib.nix b/lib.nix
index 0e61b94..e7e0648 100644
--- a/lib.nix
+++ b/lib.nix
@@ -38,18 +38,18 @@ let
     (filterAttrs (_: hostOpts: hostOpts.domain == domain) entities.hosts);
 
   getSiteGatewayV4 = siteName:
-    let site = config.fudo.sites."${siteName}";
+    let site = entities.sites."${siteName}";
     in if hasAttr "local-gateway" site then getHostIpv4 else site.gateway-v4;
 
   getHostGatewayV4 = hostname:
-    getSiteGatewayV4 config.fudo.hosts."${hostname}".site;
+    getSiteGatewayV4 entities.hosts."${hostname}".site;
 
   getSiteGatewayV6 = siteName:
-    let site = config.fudo.sites."${siteName}";
+    let site = entities.sites."${siteName}";
     in if hasAttr "local-gateway" site then getHostIpv6 else site.gateway-v6;
 
   getHostGatewayV6 = hostname:
-    getSiteGatewayV6 config.fudo.hosts."${hostname}".site;
+    getSiteGatewayV6 entities.hosts."${hostname}".site;
 
 in {
   inherit getHostSite getHostDomain getHostRealm getHostFqdn getHostIpv4