diff --git a/pkgs/development/python-modules/watermark/default.nix b/pkgs/development/python-modules/watermark/default.nix
index b6403a38dfb0..10be7466063c 100644
--- a/pkgs/development/python-modules/watermark/default.nix
+++ b/pkgs/development/python-modules/watermark/default.nix
@@ -9,28 +9,34 @@
 
 buildPythonPackage rec {
   pname = "watermark";
-  version = "2.3.0";
+  version = "2.3.1";
+  format = "setuptools";
+
   disabled = pythonOlder "3.7";
 
   src = fetchFromGitHub {
     owner = "rasbt";
     repo = pname;
-    rev = "v${version}";
-    sha256 = "1kl9yn1pkl84d3lcz7bvphqkydsgs0p5k0ja0msy3hrxxfzdzd16";
+    rev = "refs/tags/${version}";
+    hash = "sha256-E3UxdGlxTcvkiKa3RoG9as6LybyW+QrCUZvA9VHwxlk=";
   };
 
   propagatedBuildInputs = [
     ipython
-  ] ++ lib.optionals (pythonOlder "3.8") [ importlib-metadata ];
+  ] ++ lib.optionals (pythonOlder "3.8") [
+    importlib-metadata
+  ];
 
   checkInputs =  [
     pytestCheckHook
   ];
 
-  pythonImportsCheck = [ "watermark" ];
+  pythonImportsCheck = [
+    "watermark"
+  ];
 
   meta = with lib; {
-    description = "IPython extension for printing date and time stamps, version numbers, and hardware information.";
+    description = "IPython extension for printing date and timestamps, version numbers, and hardware information";
     homepage = "https://github.com/rasbt/watermark";
     license = licenses.bsd3;
     maintainers = with maintainers; [ nphilou ];