diff --git a/tools/site_editing/__init__.py b/tools/site_editing/__init__.py
index ab78603..f6e6fd7 100644
--- a/tools/site_editing/__init__.py
+++ b/tools/site_editing/__init__.py
@@ -220,8 +220,9 @@ def _factory(context: dict) -> list:
 
 site_editing_toolset = ToolSet(
     name="site_editing",
-    system_prompt=SYSTEM_PROMPT,
-    openai_schemas=TOOLS,
-    factory=_factory,
+    description="Edit Better Life SG and Memoraiz project files",
+    capability="site_editing",
+    system_prompt_fragment=SYSTEM_PROMPT,
+    build_tools=_factory,
     required_keys=[],
 )