diff --git a/kokoro/github/ubuntu/cpu/oldest/continuous.cfg b/kokoro/github/ubuntu/cpu/oldest/continuous.cfg
index 64d44859..871d566f 100644
--- a/kokoro/github/ubuntu/cpu/oldest/continuous.cfg
+++ b/kokoro/github/ubuntu/cpu/oldest/continuous.cfg
@@ -6,7 +6,7 @@ env_vars: {
 }
 env_vars: {
   key: "TF_VERSION"
-  value: "2.12.*"
+  value: "2.9.*"
 }
 env_vars: {
   key: "TAG_FILTERS"
diff --git a/kokoro/github/ubuntu/cpu/oldest/presubmit.cfg b/kokoro/github/ubuntu/cpu/oldest/presubmit.cfg
index 64d44859..871d566f 100644
--- a/kokoro/github/ubuntu/cpu/oldest/presubmit.cfg
+++ b/kokoro/github/ubuntu/cpu/oldest/presubmit.cfg
@@ -6,7 +6,7 @@ env_vars: {
 }
 env_vars: {
   key: "TF_VERSION"
-  value: "2.12.*"
+  value: "2.9.*"
 }
 env_vars: {
   key: "TAG_FILTERS"
diff --git a/setup.py b/setup.py
index 5404eab2..c5dc635a 100644
--- a/setup.py
+++ b/setup.py
@@ -168,7 +168,7 @@ def get_version():
     ],
     namespace_packages=[],
     install_requires=[
-        'apache-beam',
+        'apache-beam~=2.46.0',
         'google-vizier>=0.0.13',
         'ml-collections',
         'networkx',
@@ -177,6 +177,9 @@ def get_version():
         'tensorflow>=2.9.0; platform_machine != "arm64" or platform_system != "Darwin"',
         'tensorflow-macos>=2.9.0; platform_machine == "arm64" and platform_system == "Darwin"',
         # pylint:enable=g-line-too-long
+        # TODO(b/283835852): Remove when Vizier can upgrade to protobuf >= 4.0.
+        'protobuf~=3.20.3',
+        'wrapt~=1.14.1',
     ],
     python_requires='>=3.8,<4',
     packages=find_namespace_packages(