diff --git a/pyproject.toml b/pyproject.toml
index 4471940..71292b8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -43,7 +43,7 @@ dependencies = [
         "google-cloud-monitoring",
         "omegaconf",
         # Jetstream dependencies
-        "setuptools==61.0",
+        "setuptools==78.1.1",
         "jaxtyping",
         "clu",
         "editdistance",