diff --git a/pkgs/development/python-modules/warp-lang/default.nix b/pkgs/development/python-modules/warp-lang/default.nix
index ca97757e3ff3..d9550d87bd0f 100644
--- a/pkgs/development/python-modules/warp-lang/default.nix
+++ b/pkgs/development/python-modules/warp-lang/default.nix
@@ -43,7 +43,7 @@ let
 
   libmathdx = callPackage ./libmathdx.nix { };
 in
-buildPythonPackage {
+buildPythonPackage.override { stdenv = effectiveStdenv; } {
   pname = "warp-lang";
   inherit version;
   pyproject = true;
@@ -53,8 +53,6 @@ buildPythonPackage {
   # provided by NVCC.
   __structuredAttrs = true;
 
-  stdenv = effectiveStdenv;
-
   src = fetchFromGitHub {
     owner = "NVIDIA";
     repo = "warp";