diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 63f6cce0a6..82611559ff 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -93,7 +93,7 @@ jobs:
       - name: Checkout ${{ matrix.ref }}
         uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
         with:
-          ref: ${{ matrix.ref }}
+          ref: ${{ github.event_name == 'pull_request' && matrix.slug == 'latest' && github.event.pull_request.head.sha || matrix.ref }}
 
       - name: Resolve commit SHA
         # matrix.ref is a branch name (e.g. "releases/v0.13.0") that can move,
diff --git a/build_scripts/gen_api_md.py b/build_scripts/gen_api_md.py
index 1b9e81e84c..c1d8d1e056 100644
--- a/build_scripts/gen_api_md.py
+++ b/build_scripts/gen_api_md.py
@@ -689,7 +689,12 @@ def collect_top_level_modules(api_json_dir: Path) -> list[dict]:
     modules: list[dict] = []
     for jf in sorted(api_json_dir.glob("*.json")):
         data = json.loads(jf.read_text(encoding="utf-8"))
-        modules.extend(_expand_module(data))
+        if jf.stem.endswith("_all"):
+            submodules = [member for member in data.get("members", []) if member.get("kind") == "module"]
+            for submodule in submodules:
+                modules.extend(_expand_module(submodule))
+        else:
+            modules.extend(_expand_module(data))
 
     # Drop excluded and empty modules
     return [
diff --git a/pyrit/memory/memory_models.py b/pyrit/memory/memory_models.py
index 2cc9ef0dc7..8fc1247bfd 100644
--- a/pyrit/memory/memory_models.py
+++ b/pyrit/memory/memory_models.py
@@ -1251,7 +1251,7 @@ class EmbeddingMessageWithSimilarity(BaseModel):
     """
     Represents an embedding message with its similarity score.
 
-    Parameters:
+    Attributes:
         uuid (uuid.UUID): The UUID of the embedding message.
         metric (str): The metric used to calculate the similarity score.
         score (float): The similarity score (default is 0.0).