diff --git a/flake.nix b/flake.nix
index c6c5c83..b06e340 100644
--- a/flake.nix
+++ b/flake.nix
@@ -35,9 +35,9 @@
           pkgs = import nixpkgs {
             inherit system overlays;
           };
-          rustToolchain =
-            (pkgs.pkgsBuildHost.rust-bin.fromRustupToolchainFile ./rust-toolchain.toml).override
-              { targets = [ "wasm32-unknown-unknown" ]; };
+          rustToolchain = (pkgs.rust-bin.fromRustupToolchainFile ./rust-toolchain.toml).override {
+            targets = [ "wasm32-unknown-unknown" ];
+          };
         in
         {
           default = pkgs.mkShell {
diff --git a/nix/package.nix b/nix/package.nix
index 46eae8f..780c6dc 100644
--- a/nix/package.nix
+++ b/nix/package.nix
@@ -2,7 +2,7 @@
   hello,
   lib,
   makeRustPlatform,
-  pkgsBuildHost,
+  rust-bin,
 
   targetBins ? [ ],
   features ? [ ],
@@ -12,7 +12,7 @@
 }:
 let
   cargoToml = (lib.importTOML ../Cargo.toml);
-  rustToolchain = (pkgsBuildHost.rust-bin.fromRustupToolchainFile ../rust-toolchain.toml).override (
+  rustToolchain = (rust-bin.fromRustupToolchainFile ../rust-toolchain.toml).override (
     if cargoBuildTarget != null then
       {
         targets = [ cargoBuildTarget ];