diff --git a/pkgs/development/python-modules/ydata-profiling/default.nix b/pkgs/development/python-modules/ydata-profiling/default.nix
index 7f3a564c81c0..05c1e8d76d40 100644
--- a/pkgs/development/python-modules/ydata-profiling/default.nix
+++ b/pkgs/development/python-modules/ydata-profiling/default.nix
@@ -2,7 +2,12 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
-  pytestCheckHook,
+
+  # build-system
+  setuptools,
+  setuptools-scm,
+
+  # dependencies
   dacite,
   filetype,
   imagehash,
@@ -14,22 +19,23 @@
   numpy,
   pandas,
   phik,
-  pyarrow,
   pydantic,
   pyyaml,
   requests,
   scipy,
-  setuptools,
-  setuptools-scm,
   seaborn,
   statsmodels,
   tqdm,
   typeguard,
   visions,
   wordcloud,
+
+  # tests
+  pyarrow,
+  pytestCheckHook,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "ydata-profiling";
   version = "4.18.1";
   pyproject = true;
@@ -37,7 +43,7 @@ buildPythonPackage rec {
   src = fetchFromGitHub {
     owner = "ydataai";
     repo = "ydata-profiling";
-    tag = "v${version}";
+    tag = "v${finalAttrs.version}";
     hash = "sha256-CNeHsOpFkKvcCWGEholabcsqXJzINUUxFZ7I5bPBoYM=";
   };
 
@@ -48,7 +54,7 @@ buildPythonPackage rec {
   '';
 
   preBuild = ''
-    echo ${version} > VERSION
+    echo ${finalAttrs.version} > VERSION
   '';
 
   build-system = [
@@ -119,9 +125,9 @@ buildPythonPackage rec {
   meta = {
     description = "Create HTML profiling reports from Pandas DataFrames";
     homepage = "https://ydata-profiling.ydata.ai";
-    changelog = "https://github.com/ydataai/ydata-profiling/releases/tag/v${version}";
+    changelog = "https://github.com/ydataai/ydata-profiling/releases/tag/${finalAttrs.src.tag}";
     license = lib.licenses.mit;
     maintainers = with lib.maintainers; [ bcdarwin ];
     mainProgram = "ydata_profiling";
   };
-}
+})