diff --git a/setup.py b/setup.py
index 6aa1bc10d..a52c19f6a 100644
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,7 @@ def readme():
     ],
     # https://github.com/SeldonIO/alibi-detect/issues/375 and 387
     "tensorflow": [
-        "tensorflow_probability>=0.8.0, <0.21.0",
+        "tensorflow_probability>=0.8.0, <0.22.0",
         "tensorflow>=2.2.0, !=2.6.0, !=2.6.1, <2.14.0",  # https://github.com/SeldonIO/alibi-detect/issues/375 and 387
     ],
     "keops": [
@@ -27,7 +27,7 @@ def readme():
     ],
     "all": [
         "prophet>=1.1.0, <2.0.0",
-        "tensorflow_probability>=0.8.0, <0.21.0",
+        "tensorflow_probability>=0.8.0, <0.22.0",
         "tensorflow>=2.2.0, !=2.6.0, !=2.6.1, <2.14.0",  # https://github.com/SeldonIO/alibi-detect/issues/375 and 387
         "pykeops>=2.0.0, <2.2.0",
         "torch>=1.7.0, <1.14.0"