diff --git a/lang_agent/graphs/tool_nodes.py b/lang_agent/graphs/tool_nodes.py
index 70d427a..98f426c 100644
--- a/lang_agent/graphs/tool_nodes.py
+++ b/lang_agent/graphs/tool_nodes.py
@@ -201,7 +201,7 @@ AnnotatedToolNode = tyro.conf.OmitSubcommandPrefixes[tyro.conf.SuppressFixed[too
 
 def debug_chatty_node():
     from langchain_core.messages.base import BaseMessageChunk
-    from lang_agent.tool_manager import ToolManagerConfig
+    from lang_agent.components.tool_manager import ToolManagerConfig
     
     from dotenv import load_dotenv
     load_dotenv()