diff --git a/spacy_llm/models/bedrock/model.py b/spacy_llm/models/bedrock/model.py
index 7d09bedf..e8f12513 100644
--- a/spacy_llm/models/bedrock/model.py
+++ b/spacy_llm/models/bedrock/model.py
@@ -69,7 +69,7 @@ def __call__(self, prompts: Iterable[str]) -> Iterable[str]:
         api_responses: List[str] = []
         prompts = list(prompts)
 
-        def _request(json_data: str) -> Dict[str, Any]:
+        def _request(json_data: str) -> str:
             try:
                 import boto3
                 import botocore