diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 83dc1835..6f720480 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -23,6 +23,7 @@ jobs:
           poetry install --with test
           poetry run python -m pip install --upgrade pip
           poetry run pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+          poetry run pip install lightning
       - name: Test with pytest
         run: |
           poetry run pytest tests/