diff --git a/system/services/netbird/default.nix b/system/services/netbird/default.nix
index c2dd24f..a178d02 100644
--- a/system/services/netbird/default.nix
+++ b/system/services/netbird/default.nix
@@ -1,6 +1,6 @@
-{ ... }: {
+{ inputs, ... }: {
 
-    flake.nixosModules.default = { config, lib, nixpkgs-us, ... }: {
+    flake.nixosModules.default = { config, lib, ... }: {
 
         options.sysconfig = {
 
@@ -11,7 +11,7 @@
         };
 
         config = let 
-            pkgs-us = import nixpkgs-us {
+            pkgs-us = import inputs.nixpkgs-us {
                 system = "x86_64-linux";
             };
         in lib.mkIf config.sysconfig.services.netbird.enable {
diff --git a/system/services/ollama/default.nix b/system/services/ollama/default.nix
index 29dbeaa..062326c 100644
--- a/system/services/ollama/default.nix
+++ b/system/services/ollama/default.nix
@@ -1,6 +1,6 @@
-{ ... }: {
+{ inputs, ... }: {
 
-    flake.nixosModules.default = { config, lib, nixpkgs-us, ... }: {
+    flake.nixosModules.default = { config, lib, ... }: {
 
         options = {
             sysconfig.services.ollama.enable = lib.options.mkOption {
@@ -17,7 +17,7 @@
                     OLLAMA_CONTEXT_LENGTH = lib.mkDefault "16000";
                 };
                 package = let 
-                    pkgs-us = import nixpkgs-us { 
+                    pkgs-us = import inputs.nixpkgs-us { 
                         system = "x86_64-linux";
                         config.allowUnfree = true;
                     };