diff --git a/src/google/adk/tools/base_toolset.py b/src/google/adk/tools/base_toolset.py
index 6939953..e21d402 100644
--- a/src/google/adk/tools/base_toolset.py
+++ b/src/google/adk/tools/base_toolset.py
@@ -3,8 +3,8 @@ from abc import abstractmethod
 from typing import Optional, runtime_checkable
 from typing import Protocol
 
-from google.adk.agents.readonly_context import ReadonlyContext
-from google.adk.tools.base_tool import BaseTool
+from ..agents.readonly_context import ReadonlyContext
+from .base_tool import BaseTool
 
 
 @runtime_checkable