diff --git a/pkgs/development/python-modules/torch/tests/mk-runtime-check.nix b/pkgs/development/python-modules/torch/tests/mk-runtime-check.nix
index 61180a19aaba..3ead20d3eeb0 100644
--- a/pkgs/development/python-modules/torch/tests/mk-runtime-check.nix
+++ b/pkgs/development/python-modules/torch/tests/mk-runtime-check.nix
@@ -3,10 +3,9 @@
   feature,
   libraries,
   versionAttr,
-  pythonPackages,
 }:
 
-(cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; })
+cudaPackages.writeGpuTestPython
   {
     inherit feature;
     inherit libraries;
diff --git a/pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix b/pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix
index 268ed5297da9..9741902dab2a 100644
--- a/pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix
+++ b/pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix
@@ -4,13 +4,12 @@
   lib,
   libraries,
   name ? if feature == null then "torch-compile-cpu" else "torch-compile-${feature}",
-  pythonPackages,
   stdenv,
 }:
 let
   deviceStr = if feature == null then "" else '', device="cuda"'';
 in
-(cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; })
+cudaPackages.writeGpuTestPython
   {
     inherit name feature libraries;
     makeWrapperArgs = [