diff --git a/setup.py b/setup.py
index eee54d2..784e044 100644
--- a/setup.py
+++ b/setup.py
@@ -51,7 +51,7 @@
         'scipy == 1.7.3',
         'seqio == 0.0.8',
         't5 == 0.9.3',
-        'tensorflow',
+        'tensorflow == 2.9.1',
         'tensorflow-datasets == 4.6.0',
     ],
     classifiers=[