diff --git a/datasets/e2e/pytorch/pyproject.toml b/datasets/e2e/pytorch/pyproject.toml
index b0bb9da9a73a..4565cce9f828 100644
--- a/datasets/e2e/pytorch/pyproject.toml
+++ b/datasets/e2e/pytorch/pyproject.toml
@@ -10,7 +10,7 @@ authors = ["The Flower Authors <hello@flower.dev>"]
 
 [tool.poetry.dependencies]
 python = "^3.8"
-flwr-datasets = { path = "./../../", develop = true, extras = ["vision"] }
+flwr-datasets = { path = "./../../", extras = ["vision"] }
 torch = "^1.12.0"
 torchvision = "^0.14.1"
 parameterized = "==0.9.0"