diff --git a/pyproject.toml b/pyproject.toml
index e7805b5..6d204d4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -58,7 +58,7 @@ torchvision = "0.16.0a0"
 tornado = "<6.2"
 xarray = "2023.11.0"
 xgboost = "1.7.5"
-xpysom-dask = { git = "https://github.com/jcfaracco/xpysom-dask.git", branch = "dask" }
+xpysom-dask = { git = "https://github.com/jcfaracco/xpysom-dask.git", tag = "v1.0.7.1" }
 zarr = "2.16.1"