diff --git a/docetl/operations/utils/api.py b/docetl/operations/utils/api.py
index 997da266..7c02fb16 100644
--- a/docetl/operations/utils/api.py
+++ b/docetl/operations/utils/api.py
@@ -446,7 +446,7 @@ def _call_llm_with_cache(
             len(props) == 1
             and list(props.values())[0].get("type") == "string"
             and scratchpad is None
-            and ("ollama" in model or "sagemaker" in model)
+            and ("sagemaker" in model)
         ):
             use_tools = False
 
@@ -698,7 +698,7 @@ def _parse_llm_response_helper(
                         if key not in output_dict:
                             output_dict[key] = "Not found"
 
-                    if "ollama" in response.model:
+                    if "ollama" in response.model or "sagemaker" in response.model:
                         for key, value in output_dict.items():
                             if not isinstance(value, str):
                                 continue