diff --git a/challenge-0/infra/azuredeploy.json b/challenge-0/infra/azuredeploy.json
index ab2047c..cb78c58 100644
--- a/challenge-0/infra/azuredeploy.json
+++ b/challenge-0/infra/azuredeploy.json
@@ -302,7 +302,7 @@
       "name": "[format('{0}/{1}', variables('aiFoundryName'), 'mistral-document-ai-2505')]",
       "sku": {
         "name": "GlobalStandard",
-        "capacity": 1
+        "capacity": 60
       },
       "properties": {
         "model": {
diff --git a/challenge-6/agents/coverage_validation_agent.py b/challenge-6/agents/coverage_validation_agent.py
index 9b006ec..1dc28c1 100644
--- a/challenge-6/agents/coverage_validation_agent.py
+++ b/challenge-6/agents/coverage_validation_agent.py
@@ -144,7 +144,7 @@ def validate_coverage(claim_data: dict, policy_data: dict = None, project_client
 
             response = openai_client.responses.create(
                 input=user_query,
-                extra_body={"agent": {"name": agent.name, "type": "agent_reference"}},
+                extra_body={"agent_reference": {"name": agent.name, "type": "agent_reference"}},
             )
 
             response_text = response.output_text.strip()
diff --git a/challenge-6/agents/policy_matching_agent.py b/challenge-6/agents/policy_matching_agent.py
index df64545..c98758f 100644
--- a/challenge-6/agents/policy_matching_agent.py
+++ b/challenge-6/agents/policy_matching_agent.py
@@ -264,7 +264,7 @@ def match_policy(claim_data: dict, project_client=None) -> dict:
 
             response = openai_client.responses.create(
                 input=user_query,
-                extra_body={"agent": {"name": agent.name, "type": "agent_reference"}},
+                extra_body={"agent_reference": {"name": agent.name, "type": "agent_reference"}},
             )
 
             response_text = response.output_text.strip()