diff --git a/packages/datacommons-mcp/examples/sample_agents/basic_agent/agent.py b/packages/datacommons-mcp/examples/sample_agents/basic_agent/agent.py
index 804ebca..4ec3084 100644
--- a/packages/datacommons-mcp/examples/sample_agents/basic_agent/agent.py
+++ b/packages/datacommons-mcp/examples/sample_agents/basic_agent/agent.py
@@ -6,9 +6,8 @@
 
 from google.adk.agents.llm_agent import LlmAgent
 from google.adk.tools.mcp_tool.mcp_toolset import (
-    McpToolset,
-    StdioConnectionParams,
-    StdioServerParameters,
+    MCPToolset,
+    StreamableHTTPConnectionParams
 )
 
 from .instructions import AGENT_INSTRUCTIONS
@@ -28,14 +27,14 @@
     name="basic_agent",
     instruction=AGENT_INSTRUCTIONS,
     tools=[
-        McpToolset(
-            connection_params=StdioConnectionParams(
-                timeout=10,
-                server_params=StdioServerParameters(
-                    command="uvx",
-                    args=["datacommons-mcp", "serve", "stdio"],
-                    env={"DC_API_KEY": DC_API_KEY},
-                ),
+        MCPToolset(
+            connection_params=StreamableHTTPConnectionParams(
+                url="https://api.datacommons.org/mcp",
+                headers={
+                    "Content-Type": "application/json",
+                    "Accept": "application/json, text/event-stream",
+                    "X-API-Key": DC_API_KEY
+                }
             )
         )
     ],