diff --git a/xplainet/pyproject.toml b/xplainet/pyproject.toml
index a21dcd3..7c30292 100755
--- a/xplainet/pyproject.toml
+++ b/xplainet/pyproject.toml
@@ -13,7 +13,7 @@ exclude = []
 
 [tool.poetry.dependencies]
 python = "^3.7"
-tensorflow = "2.3.1"
+tensorflow = "2.21.0"
 tensorflow-addons = "0.11.2"
 scikit-learn = "0.23.2"