diff --git a/packages/toolbox-core/src/toolbox_core/tool.py b/packages/toolbox-core/src/toolbox_core/tool.py
index b9f5b8df..9ad420f6 100644
--- a/packages/toolbox-core/src/toolbox_core/tool.py
+++ b/packages/toolbox-core/src/toolbox_core/tool.py
@@ -26,8 +26,7 @@
 
 from aiohttp import ClientSession
 
-from toolbox_core.protocol import ParameterSchema
-
+from .protocol import ParameterSchema
 from .utils import (
     create_func_docstring,
     identify_required_authn_params,