diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 48e84a7..4a310b9 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -33,10 +33,10 @@ jobs:
       - name: Configure GitHub Pages
         uses: actions/configure-pages@v5
 
-      - name: Install package and demo dependencies
+      - name: Install package and docs dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install -e ".[demo]"
+          python -m pip install -e ".[demo,docs]"
 
       - name: Build Great Docs site
         env:
diff --git a/setup.py b/setup.py
index 102115f..989ac97 100644
--- a/setup.py
+++ b/setup.py
@@ -41,6 +41,7 @@
         "docs": [
             "pdoc>=16,<17",
             "playwright>=1.49,<2",
+            "pygments>=2.18,<3",
         ],
     },
     classifiers=[