diff --git a/pkgs/development/python-modules/torch/gpu-checks.nix b/pkgs/development/python-modules/torch/gpu-checks.nix
index d01fffe45cb0cd..55a4b45f715229 100644
--- a/pkgs/development/python-modules/torch/gpu-checks.nix
+++ b/pkgs/development/python-modules/torch/gpu-checks.nix
@@ -13,7 +13,7 @@ let
       torch,
       cudaPackages,
     }:
-    cudaPackages.writeGpuPythonTest
+    cudaPackages.writeGpuTestPython
       {
         inherit feature;
         libraries = [ torch ];