diff --git a/setup.py b/setup.py
index addb9fe..e32031d 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@
     packages=["t2ebm"],
     python_requires=">=3.6",
     install_requires=[
-        "numpy",
+        "numpy>=1.14.0,<2.0.0",
         "matplotlib",
         "tiktoken",
         "openai>=1.8.0",