diff --git a/system/pinebook.nix b/system/pinebook.nix
index 3632aac..b9d4b40 100644
--- a/system/pinebook.nix
+++ b/system/pinebook.nix
@@ -1,6 +1,4 @@
-{ lib, ...}: let
-  ips = import ./modules/ips.nix;
-in {
+{ lib, ...}: {
   imports = [
     # Base configs
     ./base.nix
@@ -46,7 +44,7 @@ in {
   hardware.opengl.driSupport32Bit = lib.mkForce false;
 
   # Set the VPN IP per machine
-  networking.wireguard.interfaces."${ips.wgInt}".ips = [ "${ips.wgSpan}.17/24" ];
+  networking.wireguard.interfaces."${outputs.ips.wgInt}".ips = [ "${outputs.ips.wgSpan}.17/24" ];
 
   # https://nixos.wiki/wiki/FAQ/When_do_I_update_stateVersion
   system.stateVersion = "24.05";