From dcdab7d01924f516decaef95a01de9e93b067076 Mon Sep 17 00:00:00 2001
From: Gabriel Arazas <foodogsquared@foodogsquared.one>
Date: Fri, 19 Jul 2024 13:48:56 +0800
Subject: [PATCH] flake-parts/setups: fix NixOS image creation function

Forgot to update this... whoops...
---
 modules/flake-parts/setups/nixos.nix | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/modules/flake-parts/setups/nixos.nix b/modules/flake-parts/setups/nixos.nix
index 8ec28c84..8fe80657 100644
--- a/modules/flake-parts/setups/nixos.nix
+++ b/modules/flake-parts/setups/nixos.nix
@@ -57,17 +57,17 @@ let
     inputs.nixos-generators.nixosModules // customFormats;
 
   # A very very thin wrapper around `mkHost` to build with the given format.
-  mkImage =
-    { system
-    , nixpkgsBranch ? "nixpkgs"
-    , extraModules ? [ ]
-    , format ? "iso"
-    }:
+  mkImage = {
+    pkgs,
+    system,
+    extraModules ? [ ],
+    format ? "iso",
+  }:
     let
       extraModules' =
         extraModules ++ [ nixosGeneratorsModulesSet.${format} ];
       image = mkHost {
-        inherit nixpkgsBranch system;
+        inherit pkgs system;
         extraModules = extraModules';
       };
     in