diff --git a/pretor/utils/get_tool.py b/pretor/utils/get_tool.py
index 8fe6fd2..b735c41 100644
--- a/pretor/utils/get_tool.py
+++ b/pretor/utils/get_tool.py
@@ -53,7 +53,7 @@ async def get_tool(agent_name: str) -> List[Callable]:
     cached = _agent_tool_result_cache.get(agent_name)
     if cached is not None:
         return cached
-    global_state_machine = ray_actor_hook("global_state_machine")
+    global_state_machine = ray_actor_hook("global_state_machine").global_state_machine
     _tool_list = await global_state_machine.get_tool_list.remote(agent_name)
     tool_list = []
     for tool_name in _tool_list.keys():