diff --git a/modules/core/nvidia.nix b/modules/core/nvidia.nix
index 9fb30ed..9c54aef 100644
--- a/modules/core/nvidia.nix
+++ b/modules/core/nvidia.nix
@@ -13,14 +13,22 @@ in
       nvidiaSettings = cfg.desktop.enable;
       # Optionally, you may need to select the appropriate driver version for your specific GPU.
       #package = config.boot.kernelPackages.nvidiaPackages.beta;
-      package = config.boot.kernelPackages.nvidiaPackages.mkDriver {
+      package =
+    let
+      drv = config.boot.kernelPackages.nvidiaPackages.mkDriver {
         version = "590.48.01";
         sha256_64bit = "sha256-ueL4BpN4FDHMh/TNKRCeEz3Oy1ClDWto1LO/LWlr1ok=";
         sha256_aarch64 = "sha256-xctt4TPRlOJ6r5S54h5W6PT6/3Zy2R4ASNFPu8TSHKM=";
         openSha256 = "sha256-ZpuVZybW6CFN/gz9rx+UJvQ715FZnAOYfHn5jt5Z2C8=";
         settingsSha256 = "sha256-ZpuVZybW6CFN/gz9rx+UJvQ715FZnAOYfHn5jt5Z2C8=";
         persistencedSha256 = lib.fakeSha256;
-      }.overrideAttrs (oldAttrs: { buildInputs = oldAttrs.buildInputs ++ [ pkgs.pkg-config ]; });
+      };
+    in
+      drv.driver.overrideAttrs (old: {
+        nativeBuildInputs =
+          (old.nativeBuildInputs or [])
+          ++ [ pkgs.pkg-config ];
+      });
     };
     hardware.nvidia-container-toolkit.enable = config.virtualisation.docker.enable;
     environment.systemPackages = with pkgs; [