diff --git a/pkgs/development/python-modules/dkimpy/default.nix b/pkgs/development/python-modules/dkimpy/default.nix
index 754e5b2f73f9..ab41cee512bb 100644
--- a/pkgs/development/python-modules/dkimpy/default.nix
+++ b/pkgs/development/python-modules/dkimpy/default.nix
@@ -4,11 +4,10 @@
   openssl,
   buildPythonPackage,
   setuptools,
-  pytest,
+  pytestCheckHook,
   dnspython,
   pynacl,
   authres,
-  python,
 }:
 
 buildPythonPackage rec {
@@ -21,8 +20,6 @@ buildPythonPackage rec {
     hash = "sha256-tfYPtHu/XY12LxNLzqDDiOumtJg0KmgqIfFoZUUJS3c=";
   };
 
-  nativeCheckInputs = [ pytest ];
-
   build-system = [ setuptools ];
 
   dependencies = [
@@ -37,9 +34,7 @@ buildPythonPackage rec {
       /usr/bin/openssl ${openssl}/bin/openssl
   '';
 
-  checkPhase = ''
-    ${python.interpreter} ./test.py
-  '';
+  nativeCheckInputs = [ pytestCheckHook ];
 
   meta = {
     description = "DKIM + ARC email signing/verification tools + Python module";