diff --git a/environment.yml b/environment.yml
index 8d6d7f9..5e77033 100644
--- a/environment.yml
+++ b/environment.yml
@@ -14,6 +14,7 @@ dependencies:
 - flox
 - fsspec
 - gdal
+- mock
 - numpy >= 1.22.0
 - rasterio >= 1.3.0
 - rioxarray
diff --git a/pyproject.toml b/pyproject.toml
index b01a509..43f2b0a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,6 +17,7 @@ classifiers = [
 dependencies = [
   "attrs",
   "flox",
+  "mock",
   "numpy",
   "pandas",
   "rasterio",