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