diff --git a/.github/workflows/reproduce_paper.yml b/.github/workflows/reproduce_paper.yml
index 8fa97ff..8098c61 100644
--- a/.github/workflows/reproduce_paper.yml
+++ b/.github/workflows/reproduce_paper.yml
@@ -36,6 +36,7 @@ jobs:
           python -m pip install -e .[molecules]
           python -m pip install scikit-learn==1.2.2
           python -m pip install notebook
+          python -m pip install "numpy<2"
       - name: Backup Reference Splits
         run: |
           cd examples/reproduce_paper_splits