diff --git a/setup.py b/setup.py
index 9060b734..1a2123a2 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.8',
     'tensorflow-privacy>=0.5,<0.8',
 ]
 other_deps = [