diff --git a/setup.py b/setup.py
index 1e06211edc..1ec996fe75 100644
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,7 @@
 
 requirements = [
     "tensorflow>=2.3.0",
-    "tensorflow-datasets>=3.2.1",
+    "tensorflow-datasets>=3.2.1,<4.0.0",
     "tensorflow-addons>=0.10.0",
     "setuptools>=47.1.1",
     "librosa>=0.7.2",