diff --git a/docs/requirements.txt b/docs/requirements.txt
index 73c9c6c..9932521 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -14,7 +14,7 @@ pyqtgraph
 scikit_learn
 scikit-image
 tqdm
-pywt
+PyWavelets
 trackpy
 joblib
 psutil
diff --git a/pyproject.toml b/pyproject.toml
index a844400..b8e87ee 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,6 +50,7 @@ dependencies = [
     "tqdm",
     "trackpy",
     "wget",
+    "PyWavelets",
 ]
 
 [project.optional-dependencies]
diff --git a/requirements.txt b/requirements.txt
index 755a35b..bddf550 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -27,3 +27,4 @@ tifffile
 tqdm
 trackpy
 wget
+PyWavelets
diff --git a/setup.py b/setup.py
index 4f99710..d5be33f 100644
--- a/setup.py
+++ b/setup.py
@@ -71,6 +71,7 @@
         "tensorflow",
         "GitPython",
         "subprocess.run",
+        "PyWavelets",
     ],
     # not to be confused with definitions in pyproject.toml [build-system]
     setup_requires=["pytest-runner"],