diff --git a/tests/client-sdk/agents/test_agents.py b/tests/client-sdk/agents/test_agents.py
index 4f3fda8c3..12455b066 100644
--- a/tests/client-sdk/agents/test_agents.py
+++ b/tests/client-sdk/agents/test_agents.py
@@ -165,7 +165,7 @@ def test_builtin_tool_brave_search(llama_stack_client, agent_config):
         messages=[
             {
                 "role": "user",
-                "content": "Search the web and tell me who the 44th president of the United States was.",
+                "content": "Search the web and tell me who the 44th president of the United States was. Please use tools",
             }
         ],
         session_id=session_id,
diff --git a/tests/client-sdk/inference/test_inference.py b/tests/client-sdk/inference/test_inference.py
index ea9cfb8ae..97b26c539 100644
--- a/tests/client-sdk/inference/test_inference.py
+++ b/tests/client-sdk/inference/test_inference.py
@@ -56,7 +56,7 @@ def test_image_chat_completion(llama_stack_client):
         "content": [
             {
                 "type": "image",
-                "data": {
+                "url": {
                     "uri": "https://www.healthypawspetinsurance.com/Images/V3/DogAndPuppyInsurance/Dog_CTA_Desktop_HeroImage.jpg"
                 },
             },