diff --git a/requirements.txt b/requirements.txt
index 18ec4cfe..7ab361b5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,4 +7,4 @@ sentencepiece
 tensorflow-gpu==1.15
 tensorflow-text==1.15.0rc0
 tensor2tensor==1.15.0
-tfds-nightly
+tensorflow-datasets==2.1.0