diff --git a/package.nix b/package.nix
index 2a70eab..28a0d59 100644
--- a/package.nix
+++ b/package.nix
@@ -1,8 +1,11 @@
-{ stdenvNoCC, lib }:
+{ stdenvNoCC, makeWrapper, lib, path }:
 
 stdenvNoCC.mkDerivation rec {
   name = "disko";
   src = ./.;
+  nativeBuildInputs = [
+    makeWrapper
+  ];
   installPhase = ''
     mkdir -p $out/bin $out/share/disko
     cp -r cli.nix default.nix types.nix disk-deactivate types $out/share/disko
@@ -11,6 +14,7 @@ stdenvNoCC.mkDerivation rec {
       -e "s|#!/usr/bin/env.*|#!/usr/bin/env bash|" \
       disko > $out/bin/disko
     chmod 755 $out/bin/disko
+    wrapProgram $out/bin/disko --prefix NIX_PATH : "nixpkgs=${path}"
   '';
   meta = with lib; {
     description = "Format disks with nix-config";