diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
index 835d93d4b..ce3c80beb 100644
--- a/.github/workflows/run_tests.yml
+++ b/.github/workflows/run_tests.yml
@@ -20,7 +20,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install pytest Cython fury
+          pip install pytest Cython fury xgboost
           pip install torch==2.1.0 -f https://download.pytorch.org/whl/cpu
           pip install .