diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index a6a004a1d6b330..ce5aa3be2ec3e9 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -6584,10 +6584,9 @@ self: super: with self; {
 
   jaxlib-bin = callPackage ../development/python-modules/jaxlib/bin.nix { };
 
-  jaxlib-build = callPackage ../development/python-modules/jaxlib rec {
-    # Some platforms don't have `cudaSupport` defined, hence the need for 'or false'.
+  jaxlib-build = callPackage ../development/python-modules/jaxlib {
     inherit (pkgs.config) cudaSupport;
-    IOKit = pkgs.darwin.apple_sdk_11_0.IOKit;
+    inherit (pkgs.darwin.apple_sdk_11_0) IOKit;
   };
 
   jaxlib = jaxlib-bin;