diff --git a/setup.py b/setup.py
index 9060b734..feb6f746 100644
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,7 @@
 
 keras_deps = [
     'tensorflow>=2.10',
-    'tensorflow_datasets>=4.2,<4.5',
+    'tensorflow_datasets>=4.2,<4.9',
     'tensorflow-privacy>=0.5,<0.8',
 ]
 other_deps = [