diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 4ca3500..6a51708 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -25,6 +25,7 @@ jobs:
       uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
+        cache: "pip"
     - name: Configure python
       run: |
         python -m pip install --upgrade pip
@@ -39,13 +40,10 @@ jobs:
         flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
         # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
         flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
-    - name: Test with pytest
+    - name: Test with pytest and create coverage report
       run: |
-        pytest
-        - name: Test with pytest and create coverage report
-        run: |
-          python -m poetry run coverage run --source=TPTBox -m pytest
-          python -m poetry run coverage xml
+        python -m poetry run coverage run --source=spineps -m pytest
+        python -m poetry run coverage xml
     - name: Upload coverage results to Codecov (Only on merge to main)
       # Only upload to Codecov after a merge to the main branch
       if: github.ref == 'refs/heads/main' && github.event_name == 'push'
diff --git a/pyproject.toml b/pyproject.toml
index 7d40b1a..a7aedb1 100755
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ torchmetrics = "^1.1.2"
 tqdm = "^4.66.1"
 einops= "^0.6.1"
 nnunetv2 = "2.4.2"
-tptbox = "^0.2.1"
+TPTBox = "^0.2.1"
 antspyx = "0.4.2"
 rich = "^13.6.0"
 
@@ -39,6 +39,14 @@ rich = "^13.6.0"
 [tool.poetry.dev-dependencies]
 pytest = "^7.4.4"
 pre-commit = "*"
+coverage = ">=7.0.1"
+pytest-mock = "^3.6.0"
+pandas = "^2.1.0"
+joblib = "^1.3.2"
+future = "^0.18.3"
+flake8 = ">=4.0.1"
+auxiliary = ">=0.1.0"
+tqdm = ">=4.62.3"
 
 
 [tool.poetry-dynamic-versioning]