diff --git a/setup.py b/setup.py
index c7e8c536..7e8ed428 100644
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,7 @@
 import setuptools
 
 keras_deps = [
-    'tensorflow>=2.2,<2.8',
+    'tensorflow>=2.9',
     'tensorflow_datasets>=4.2,<4.5',
     'tensorflow-privacy>=0.5,<0.8',
 ]
@@ -41,6 +41,7 @@
     "mkdocs-material",
     "mkdocs-material-extensions",
     "markdown-include",
+    "jinja2==3.0.3"
 ]
 
 grpc_deps = ['grpcio>=1.35,<1.43',
@@ -82,7 +83,7 @@
         'numpy>=1.16,<1.23',
         'pydantic>=1.7,<1.10',
     ],
-    tests_require=["tox>=3.20,<3.25"],
+    tests_require=["tox>=3.20,<3.26"],
     extras_require={
         'keras': keras_deps,
         'other': other_deps,