diff --git a/nodes/tools/gtUIConvertAgentToTool.py b/nodes/tools/gtUIConvertAgentToTool.py
index 7047191..495405f 100644
--- a/nodes/tools/gtUIConvertAgentToTool.py
+++ b/nodes/tools/gtUIConvertAgentToTool.py
@@ -46,7 +46,7 @@ class gtUIConvertAgentToTool(gtUIBaseTool):
     ):
         if agent:
             # Create a local structure function
-            driver = LocalStructureRunDriver(structure_factory_fn=lambda: agent)
+            driver = LocalStructureRunDriver(create_structure=lambda: agent)
             tool = StructureRunTool(
                 name=to_pascal_case(name),
                 # name=name,