diff --git a/.github/workflows/pr_workflow.yaml b/.github/workflows/pr_workflow.yaml
index 17a3ac20..f46b1954 100644
--- a/.github/workflows/pr_workflow.yaml
+++ b/.github/workflows/pr_workflow.yaml
@@ -19,7 +19,6 @@ jobs:
           python -m pip install --upgrade pip
           pip install wheel
           pip install numpy cython h5py
-          pip install xtrack xpart xfields nafflib
           pip install --no-build-isolation .[tests]
       - name: Run tests
         run: |
diff --git a/setup.py b/setup.py
index 40a9b861..66500fcd 100644
--- a/setup.py
+++ b/setup.py
@@ -44,9 +44,11 @@
         'numpy>=1.0',
         'scipy',
         'pyyaml',
-        'pandas'
+        'pandas',
+        'xpart',
+        'xfields'
         ],
     extras_require={
-        'tests': ['pytest', 'PyHEADTAIL'],
+        'tests': ['pytest', 'PyHEADTAIL', 'nafflib'],
         },
     )