diff --git a/pkgs/by-name/di/distgen/package.nix b/pkgs/by-name/di/distgen/package.nix
index fc2f68fdf22f..2cf615a79017 100644
--- a/pkgs/by-name/di/distgen/package.nix
+++ b/pkgs/by-name/di/distgen/package.nix
@@ -6,32 +6,44 @@
 
 python3.pkgs.buildPythonApplication rec {
   pname = "distgen";
-  version = "1.18";
+  version = "2.1";
+  pyproject = true;
 
   src = fetchPypi {
     inherit pname version;
-    hash = "sha256-lS6OeEaPiK8Pskuoww9KwyNhKnGQ+dHhdPmZn1Igj0Q=";
+    hash = "sha256-VG9EX9LHoZamBM3PEm5qGpViK39qD+PA8vcHTzvsW+o=";
   };
 
-  nativeCheckInputs = with python3.pkgs; [
-    pytest
-    mock
+  build-system = with python3.pkgs; [
+    setuptools
+    argparse-manpage
   ];
 
-  propagatedBuildInputs = with python3.pkgs; [
+  dependencies = with python3.pkgs; [
     distro
     jinja2
     six
     pyyaml
   ];
 
-  checkPhase = "make test-unit PYTHON=${python3.executable}";
+  nativeCheckInputs = with python3.pkgs; [
+    pytest
+    mock
+  ];
 
-  meta = with lib; {
+  checkPhase = ''
+    runHook preCheck
+
+    make test-unit PYTHON=${python3.executable}
+
+    runHook postCheck
+  '';
+
+  meta = {
     description = "Templating system/generator for distributions";
     mainProgram = "dg";
-    license = licenses.gpl2Plus;
-    homepage = "https://distgen.readthedocs.io/";
-    maintainers = with maintainers; [ bachp ];
+    license = lib.licenses.gpl2Plus;
+    homepage = "https://distgen.readthedocs.io";
+    maintainers = with lib.maintainers; [ bachp ];
   };
 }