diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 0cd88ba..33d3da0 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -42,8 +42,9 @@ jobs:
       uses: ammaraskar/sphinx-action@master
       with:
         pre-build-command: |
+          apt-get update -y && apt-get install -y git
           python -m pip install --upgrade pip setuptools wheel
-          python -m pip install -e .[docs]
+          python -m pip install -e .[composable,docs]
         docs-folder: "docs/"
     - name: Publish documentation
       run: |
diff --git a/setup.py b/setup.py
index 5c0dd2d..23aa3ca 100644
--- a/setup.py
+++ b/setup.py
@@ -127,7 +127,7 @@ setup(
        'bayes': ['jax', 'jaxlib', 'numpyro'],
        'composable': ['qunfold @ git+https://github.com/mirkobunse/qunfold@v0.1.3'],
        'tests': ['certifi'],
-       'docs' : ['sphinx-rtd-theme', 'myst-parser', 'qunfold @ git+https://github.com/mirkobunse/qunfold@v0.1.3'],
+       'docs' : ['sphinx-rtd-theme', 'myst-parser'],
     },
 
     # If there are data files included in your packages that need to be