diff --git a/tests/baselines/api/stats/stats.json b/tests/baselines/api/stats/stats.json
index de5499f37..e69454805 100644
--- a/tests/baselines/api/stats/stats.json
+++ b/tests/baselines/api/stats/stats.json
@@ -430,6 +430,7 @@
         "talents": false,
         "transcripts": false
       },
+      "max_output_tokens": 1024,
       "mtime": 0,
       "output": "json",
       "path": "<PROJECT>/solstone/apps/timeline/talent/segment_summary.md",
diff --git a/tests/test_describe_capacity_retry.py b/tests/test_describe_capacity_retry.py
index 1942c2235..72e493b80 100644
--- a/tests/test_describe_capacity_retry.py
+++ b/tests/test_describe_capacity_retry.py
@@ -65,11 +65,18 @@ def _assert_no_describe_temp(directory: Path) -> None:
 
 def _capacity_error():
     from solstone.think.providers import local as local_provider
+    from solstone.think.providers.local_endpoint import LocalEndpoint
 
     inner = type("ReadTimeout", (Exception,), {})("capacity wait timed out")
     outer = RuntimeError("outer")
     outer.__cause__ = inner
-    return local_provider._classify_byo_generate_error(outer)
+    endpoint = LocalEndpoint(
+        base_url="http://endpoint.test",
+        served_model_id="m",
+        credential=None,
+        is_bundled=False,
+    )
+    return local_provider._classify_byo_generate_error(outer, endpoint)
 
 
 class RetryBatch: