diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 9b3f8fb..d700849 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -30,7 +30,7 @@ jobs:
           
       - name: Install dependencies
         run: |
-          pip install torch==2.4.0 packaging
+          pip install torch==2.4.0 packaging wheel
           pip install flash-attn==2.6.3 --no-build-isolation
           pip install .
           pip install -r requirements-dev.txt