diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml
index bd9303c368..fc039ea393 100644
--- a/.github/workflows/docs.yaml
+++ b/.github/workflows/docs.yaml
@@ -17,9 +17,9 @@ jobs:
       - name: Install dstack
         run: |
           if [ -n "${{ inputs.release_tag }}" ]; then
-            pip install "dstack==${{ inputs.release_tag }}"
+            pip install "dstack[server]==${{ inputs.release_tag }}"
           else
-            pip install -e .
+            pip install -e .[server]
           fi
       - name: Build
         run: |