diff --git a/setup.py b/setup.py
index 5939fe3..d5aa105 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@
     tests_require=tests_require,
     test_suite="modnet.tests",
     extras_require={
-        "bayesian": ["tensorflow-probability~=0.18"],
+        "bayesian": ["tensorflow-probability==0.18"],
         "test": tests_require,
         "dev": dev_require,
     },