diff --git a/setup.py b/setup.py
index 918d4b5c..bb24829b 100644
--- a/setup.py
+++ b/setup.py
@@ -40,13 +40,13 @@
         'rouge-score',
         'sacrebleu',
         'sentencepiece',
-        'tensorflow==1.15',
+        'tensorflow==2.11.1',
         'tensorflow-text==1.15.0rc0',
         'tfds-nightly',
         'tensor2tensor==1.15.0',
     ],
     extras_require={
-        'tensorflow': ['tensorflow==1.15'],
+        'tensorflow': ['tensorflow==2.11.1'],
     },
     classifiers=[
         'Development Status :: 4 - Beta',