diff --git a/setup.py b/setup.py
index 9233b4d..5b0c811 100644
--- a/setup.py
+++ b/setup.py
@@ -33,7 +33,7 @@
     "pyspark>=3.3.0",
     "delta-spark>=2.2.0",
     "pandas<2.0.0",  # From 2.0.0 onwards, pandas does not support iteritems() anymore, spark.createDataFrame will fail
-    "numpy<1.24",  # From 1.24 onwards, module 'numpy' has no attribute 'bool'.
+    "numpy<2.1",  # From 1.24 onwards, module 'numpy' has no attribute 'bool'.
 ]
 
 TEST_REQUIREMENTS = [