diff --git a/pkgs/development/python-modules/niaaml/default.nix b/pkgs/development/python-modules/niaaml/default.nix
index edb64800c378..25370ff81e3f 100644
--- a/pkgs/development/python-modules/niaaml/default.nix
+++ b/pkgs/development/python-modules/niaaml/default.nix
@@ -11,6 +11,8 @@
   scikit-learn,
   toml-adapt,
   typer,
+  typing-extensions,
+  writableTmpDirAsHomeHook,
 }:
 
 buildPythonPackage rec {
@@ -43,6 +45,7 @@ buildPythonPackage rec {
     pandas
     scikit-learn
     typer
+    typing-extensions
   ];
 
   # create scikit-learn and niapy deps version consistent
@@ -51,7 +54,10 @@ buildPythonPackage rec {
     toml-adapt -path pyproject.toml -a change -dep niapy -ver X
   '';
 
-  nativeCheckInputs = [ pytestCheckHook ];
+  nativeCheckInputs = [
+    pytestCheckHook
+    writableTmpDirAsHomeHook
+  ];
 
   pythonImportsCheck = [ "niaaml" ];