diff --git a/pkgs/development/python-modules/array-api-compat/default.nix b/pkgs/development/python-modules/array-api-compat/default.nix
index 3f25d7763b9e..6de3c073b43a 100644
--- a/pkgs/development/python-modules/array-api-compat/default.nix
+++ b/pkgs/development/python-modules/array-api-compat/default.nix
@@ -14,7 +14,6 @@
   config,
   cudaSupport ? config.cudaSupport,
   cupy,
-  nix-update-script,
 }:
 
 buildPythonPackage rec {
@@ -45,13 +44,10 @@ buildPythonPackage rec {
   pythonImportsCheck = [ "array_api_compat" ];
 
   # CUDA (used via cupy) is not available in the testing sandbox
-  checkPhase = ''
-    runHook preCheck
-    python -m pytest -k 'not cupy'
-    runHook postCheck
-  '';
-
-  passthru.updateScript = nix-update-script { };
+  pytestFlagsArray = [
+    "-k"
+    "'not cupy'"
+  ];
 
   meta = {
     homepage = "https://data-apis.org/array-api-compat";