diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 7acc7a5..4d3b356 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -12,17 +12,16 @@ jobs:
       matrix:
         python-version: ["3.8", "3.10"]
         torch-version: ["1.12", "2.0"]
-
     steps:
-      - uses: actions/checkout@v3
-      - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v4
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip setuptools
           pip install pytest
+          pip install numpy==1.*
           pip install torch==${{ matrix.torch-version }} --extra-index-url https://download.pytorch.org/whl/cpu
           pip install .
       - name: Test with pytest