diff --git a/modules/flake-parts/setups/nixos.nix b/modules/flake-parts/setups/nixos.nix
index b93e9397..c384257f 100644
--- a/modules/flake-parts/setups/nixos.nix
+++ b/modules/flake-parts/setups/nixos.nix
@@ -388,8 +388,11 @@ in
                 lib.nameValuePair
                   "${name}-${format}"
                   (mkImage {
-                    inherit (metadata) nixpkgsBranch;
-                    inherit system format;
+                    inherit format system;
+                    pkgs = import inputs.${metadata.nixpkgs.branch} {
+                      inherit system;
+                      inherit (metadata.nixpkgs) config;
+                    };
                     extraModules = cfg.sharedModules ++ metadata.modules;
                   });