diff --git a/setup.py b/setup.py
index e662f5a8..0417bed3 100644
--- a/setup.py
+++ b/setup.py
@@ -26,6 +26,7 @@
     "pre-commit",
     "pyarrow",
     "pytest",
+    "pytest-mock",
     "rasterio",
     "scikit-image",
     "scikit-learn",