diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 03880b70..b4330e8a 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -28,6 +28,6 @@ jobs:
           path: .cache
           restore-keys: |
             mkdocs-material-
-      - run: pip install mkdocs-material mkdocstrings
+      - run: pip install -r requirements.txt
       - run: pip install -e .
       - run: mkdocs gh-deploy --force
diff --git a/docs/README.md b/docs/README
similarity index 100%
rename from docs/README.md
rename to docs/README
diff --git a/src/continuity/model/neuraloperator.py b/src/continuity/model/neuraloperator.py
index d7c0c8bc..cdc3b188 100644
--- a/src/continuity/model/neuraloperator.py
+++ b/src/continuity/model/neuraloperator.py
@@ -81,8 +81,8 @@ def __init__(
             coordinate_dim: Dimension of coordinate space
             num_channels: Number of channels
             depth: Number of hidden layers
-            width: Width of kernel network
-            depth: Depth of kernel network
+            kernel_width: Width of kernel network
+            kernel_depth: Depth of kernel network
         """
         super().__init__()