diff --git a/oss/build_whl.sh b/oss/build_whl.sh
index f4fed53..e0dc052 100755
--- a/oss/build_whl.sh
+++ b/oss/build_whl.sh
@@ -92,7 +92,7 @@ function main() {
   $PYTHON_BIN -c 'from array_record.python import array_record_data_source'
   # TF is not available on Python 3.13 and above.
   if [ "$(uname)" != "Darwin" ] && (( "${PYTHON_MINOR_VERSION}" < 13 )); then
-    $PYTHON_BIN -m pip install jax tensorflow>=2.20.0rc0 grain
+    $PYTHON_BIN -m pip install jax tensorflow>=2.20.0 grain
     $PYTHON_BIN oss/test_import_grain.py
     $PYTHON_BIN oss/test_import_tensorflow.py
   fi
diff --git a/setup.py b/setup.py
index fba6a34..435c165 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
     'etils[epath]',
 ]
 
-TF_PACKAGE = ['tensorflow>=2.20.0rc0']
+TF_PACKAGE = ['tensorflow>=2.20.0']
 
 BEAM_EXTRAS = [
     'apache-beam[gcp]>=2.53.0',