diff --git a/pkgs/by-name/wh/whatstyle/package.nix b/pkgs/by-name/wh/whatstyle/package.nix
index 7002bcf7c055..e0c5f90b5c5f 100644
--- a/pkgs/by-name/wh/whatstyle/package.nix
+++ b/pkgs/by-name/wh/whatstyle/package.nix
@@ -1,6 +1,6 @@
 {
   lib,
-  python3,
+  python3Packages,
   fetchFromGitHub,
   llvmPackages,
 }:
@@ -9,21 +9,21 @@ let
   inherit (llvmPackages) clang-unwrapped;
 in
 
-python3.pkgs.buildPythonApplication rec {
+python3Packages.buildPythonApplication (finalAttrs: {
   pname = "whatstyle";
   version = "0.2.0";
   format = "setuptools";
   src = fetchFromGitHub {
     owner = "mikr";
     repo = "whatstyle";
-    tag = "v${version}";
+    tag = "v${finalAttrs.version}";
     hash = "sha256-4LCZAEUQFPl4CBPeuqsodiAlwd8uBg+SudF5d+Vz4Gc=";
   };
 
   # Fix references to previous version, to avoid confusion:
   postPatch = ''
-    substituteInPlace setup.py --replace-fail 0.1.9 ${version}
-    substituteInPlace whatstyle.py --replace-fail 0.1.9 ${version}
+    substituteInPlace setup.py --replace-fail 0.1.9 ${finalAttrs.version}
+    substituteInPlace whatstyle.py --replace-fail 0.1.9 ${finalAttrs.version}
   '';
 
   nativeCheckInputs = [
@@ -40,4 +40,4 @@ python3.pkgs.buildPythonApplication rec {
     maintainers = [ ];
     platforms = lib.platforms.all;
   };
-}
+})