diff --git a/pkgs/development/python-modules/ansi2html/default.nix b/pkgs/development/python-modules/ansi2html/default.nix
index 5387e93be6c..76775acb163 100644
--- a/pkgs/development/python-modules/ansi2html/default.nix
+++ b/pkgs/development/python-modules/ansi2html/default.nix
@@ -1,4 +1,4 @@
-{ lib, buildPythonPackage, fetchPypi, isPy3k, six, mock, nose, setuptools }:
+{ lib, buildPythonPackage, fetchPypi, isPy3k, six, mock, pytestCheckHook, setuptools, setuptools_scm, toml }:
 
 buildPythonPackage rec {
   pname = "ansi2html";
@@ -9,9 +9,13 @@ buildPythonPackage rec {
     sha256 = "0f124ea7efcf3f24f1f9398e527e688c9ae6eab26b0b84e1299ef7f94d92c596";
   };
 
+  nativeBuildInputs = [ setuptools_scm toml ];
   propagatedBuildInputs = [ six setuptools ];
 
-  checkInputs = [ mock nose ];
+  preCheck = "export PATH=$PATH:$out/bin";
+  checkInputs = [ mock pytestCheckHook ];
+
+  pythonImportsCheck = [ "ansi2html" ];
 
   meta = with lib; {
     description = "Convert text with ANSI color codes to HTML";