diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 848ae40c..0c7365cf 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -403,7 +403,8 @@ RUN pip install annoy \
         google-cloud-bigquery \
         google-cloud-storage && \
     # Split these installations to avoid `pip._vendor.resolvelib.resolvers.ResolutionTooDeep: 200000`
-    pip install google-cloud-translate==3.* \
+    # TODO(b/315753846) Unpin translate package.
+    pip install google-cloud-translate==3.12.1 \
         google-cloud-language==2.* \
         google-cloud-videointelligence==2.* \
         google-cloud-vision==2.* \
diff --git a/tests/test_keras_core.py b/tests/test_keras_core.py
index bedeefb0..29c8cd90 100644
--- a/tests/test_keras_core.py
+++ b/tests/test_keras_core.py
@@ -3,7 +3,7 @@
 import numpy as np
 import os
 
-os.environ["KERAS_BACKEND"] = "jax" 
+os.environ["KERAS_BACKEND"] = "tensorflow" 
 
 # Note that keras_core should only be imported after the backend
 # has been configured. The backend cannot be changed once the