diff --git a/pyproject.toml b/pyproject.toml
index 5252b40e..8ad8c14c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,7 +10,7 @@ dependencies = [
     "ase>=3.23.0",
     "cython>=3",
     # "numpy>=1.26",  # TODO: remove after test
-    "numpy>=2.0.0",
+    "numpy==1.26",
     "nvidia-ml-py3>=7.352.0",
     "pymatgen>=2024.9.10",
     "torch>=2.4.1",