diff --git a/default.nix b/default.nix
index 020afaf..80e1f20 100644
--- a/default.nix
+++ b/default.nix
@@ -1,6 +1,7 @@
 { system ? builtins.currentSystem,
   nvidiaVersion ? null,
-  nvidiaHash ? null
+  nvidiaHash ? null,
+  pkgs ? import <nixpkgs> { inherit system; }
 }:
 
 with pkgs;