diff --git a/setup.py b/setup.py
index 9bc6b96c..be2d18a1 100644
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,7 @@
 
 keras_deps = [
     'tensorflow>=2.9',
-    'tensorflow_datasets>=4.2,<4.5',
+    'tensorflow_datasets>=4.2,<4.7',
     'tensorflow-privacy>=0.5,<0.8',
 ]
 other_deps = [