diff --git a/pkgs/development/python-modules/pygments-better-html/default.nix b/pkgs/development/python-modules/pygments-better-html/default.nix
index d58dabf269a1..4656b8a28168 100644
--- a/pkgs/development/python-modules/pygments-better-html/default.nix
+++ b/pkgs/development/python-modules/pygments-better-html/default.nix
@@ -1,35 +1,41 @@
 {
   lib,
   buildPythonPackage,
-  fetchPypi,
-  setuptools,
+  fetchFromGitHub,
+  flit-core,
   pygments,
+  python,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "pygments-better-html";
-  version = "0.1.5";
+  version = "0.1.6";
   pyproject = true;
 
-  src = fetchPypi {
-    pname = "pygments_better_html";
-    inherit version;
-    hash = "sha256-SLAe5ubIGEchUNoHCct6CWisBja3WNEfpE48v9CTzPQ=";
+  src = fetchFromGitHub {
+    owner = "Kwpolska";
+    repo = "pygments_better_html";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-7vX/xm1lb89YLuDJmgdDCg+/UHinQAchi8OaF9TXRJA=";
   };
 
-  build-system = [ setuptools ];
+  build-system = [ flit-core ];
 
   dependencies = [ pygments ];
 
-  # has no tests
-  doCheck = false;
+  checkPhase = ''
+    runHook preCheck
+    ${python.interpreter} demo.py
+    runHook postCheck
+  '';
 
   pythonImportsCheck = [ "pygments_better_html" ];
 
   meta = {
+    changelog = "https://github.com/Kwpolska/pygments_better_html/blob/${finalAttrs.src.tag}/CHANGELOG.md";
     homepage = "https://github.com/Kwpolska/pygments_better_html";
     description = "Improved line numbering for Pygments’ HTML formatter";
     license = lib.licenses.bsd3;
     maintainers = with lib.maintainers; [ hexa ];
   };
-}
+})