diff --git a/setup.py b/setup.py
index d26e82c..15c954f 100644
--- a/setup.py
+++ b/setup.py
@@ -70,8 +70,9 @@ def read_file(filename: str) -> str:
           "yfinance==0.2.28",
           "tqdm",
           "dtaidistance >= 2.3.10",
-          "tensorflow",
-          "tensorflow-probability",
+          "tensorflow < 2.16",
+          "tensorflow-probability < 0.24.0",
+          "statsmodels"
       ],
       package_data={'tsgm': ['README.md']},
       packages=find_packages())