diff --git a/seqio/version.py b/seqio/version.py
index 3df06665..5cae9578 100644
--- a/seqio/version.py
+++ b/seqio/version.py
@@ -18,4 +18,4 @@
 pulling in all the dependencies in __init__.py.
 """
 
-__version__ = '0.0.18'
+__version__ = '0.0.20a'
diff --git a/setup.py b/setup.py
index 6c980f7e..03d7d86e 100644
--- a/setup.py
+++ b/setup.py
@@ -54,9 +54,7 @@
         'pyglove',
         'sentencepiece >= 0.2',
         'tensorflow-text',
-        # Ping to a specific version to avoid endless backtracking during
-        # pip dependency resolution.
-        'tfds-nightly==4.9.2.dev202308090034',
+        'tensorflow-datasets',
     ],
     requires_python='>=3.9',
     extras_require={