diff --git a/contributing/samples/toolbox_agent/agent.py b/contributing/samples/toolbox_agent/agent.py
index 37a5997..e7b04b1 100644
--- a/contributing/samples/toolbox_agent/agent.py
+++ b/contributing/samples/toolbox_agent/agent.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 from google.adk.agents import Agent
-from google.adk.tools import ToolboxToolset
+from google.adk.tools.toolbox_toolset import ToolboxToolset
 
 root_agent = Agent(
     model="gemini-2.0-flash",