diff --git a/setup.py b/setup.py
index c31815a..2acc65d 100644
--- a/setup.py
+++ b/setup.py
@@ -120,7 +120,7 @@ def extend_package(path):
     install_requires=[
         "pynq>=2.5.1",
         "bitstring>=3.1.7",
-        "numpy<=1.24.1",
+        "numpy<=2.0",
         "finn-dataset_loading==0.0.5",  # noqa
     ],
     extras_require={