diff --git a/pkgs/development/python-modules/sphinxcontrib-autoapi/default.nix b/pkgs/development/python-modules/sphinxcontrib-autoapi/default.nix
index 55feba377d5e..26b50d1cb773 100644
--- a/pkgs/development/python-modules/sphinxcontrib-autoapi/default.nix
+++ b/pkgs/development/python-modules/sphinxcontrib-autoapi/default.nix
@@ -1,45 +1,52 @@
-{ stdenv
-, lib
+{ lib
+, astroid
 , buildPythonPackage
 , fetchPypi
-, pythonOlder
-, astroid
 , jinja2
-, sphinx
-, pyyaml
-, unidecode
 , mock
-, pytest
+, pytestCheckHook
+, pythonOlder
+, pyyaml
+, sphinx
+, stdenv
+, unidecode
 }:
 
 buildPythonPackage rec {
   pname = "sphinx-autoapi";
   version = "1.9.0";
-  disabled = pythonOlder "3.6";
+  format = "setuptools";
+
+  disabled = pythonOlder "3.7";
 
   src = fetchPypi {
     inherit pname version;
-    sha256 = "sha256-yJfqM33xatDN4wfL3+K+ziB3iN3hWH+k/IuFfR/F3Lo=";
+    hash = "sha256-yJfqM33xatDN4wfL3+K+ziB3iN3hWH+k/IuFfR/F3Lo=";
   };
 
-  propagatedBuildInputs = [ astroid jinja2 pyyaml sphinx unidecode ];
+  propagatedBuildInputs = [
+    astroid
+    jinja2
+    pyyaml
+    sphinx
+    unidecode
+  ];
 
   checkInputs = [
     mock
-    pytest
+    pytestCheckHook
   ];
 
-  checkPhase = ''
-    pytest
-  '';
+  pythonImportsCheck = [
+    "autoapi"
+  ];
 
   meta = with lib; {
-    broken = stdenv.isDarwin;
     homepage = "https://github.com/readthedocs/sphinx-autoapi";
     description = "Provides 'autodoc' style documentation";
     longDescription = "Sphinx AutoAPI provides 'autodoc' style documentation for multiple programming languages without needing to load, run, or import the project being documented.";
     license = licenses.mit;
     maintainers = with maintainers; [ karolchmist ];
+    broken = stdenv.isDarwin;
   };
-
 }