diff --git a/pkgs/development/python-modules/nifty8/default.nix b/pkgs/development/python-modules/nifty8/default.nix
index ba3e0f14932e..70e9140123db 100644
--- a/pkgs/development/python-modules/nifty8/default.nix
+++ b/pkgs/development/python-modules/nifty8/default.nix
@@ -1,12 +1,9 @@
 {
-  config,
   lib,
   stdenv,
   buildPythonPackage,
   fetchFromGitLab,
 
-  cudaSupport ? config.cudaSupport,
-
   # build-system
   setuptools,
 
@@ -48,7 +45,7 @@ buildPythonPackage rec {
     ducc0
     h5py
     jax
-    (jaxlib.override { inherit cudaSupport; })
+    jaxlib
     matplotlib
     mpi4py
     mpi