diff --git a/vlmrun/client/agent.py b/vlmrun/client/agent.py
index 3c02411..7f859f4 100644
--- a/vlmrun/client/agent.py
+++ b/vlmrun/client/agent.py
@@ -183,7 +183,7 @@ def execute(
             callback_url: Optional URL to call when execution is complete
             model: VLM Run Agent model to use for execution (default: "vlmrun-orion-1:auto")
             toolsets: Optional list of tool categories to enable for this execution.
-                Available categories: core, image, image-gen, 3d_reconstruction,
+                Available categories: core, image, image-gen, world_gen,
                 viz, document, video, web.
                 When specified, only tools from these categories will be available.
                 If None, defaults to 'core' tools only.
diff --git a/vlmrun/client/types.py b/vlmrun/client/types.py
index bc869c3..48329c0 100644
--- a/vlmrun/client/types.py
+++ b/vlmrun/client/types.py
@@ -17,7 +17,7 @@
     "core",
     "image",
     "image-gen",
-    "3d_reconstruction",
+    "world_gen",
     "viz",
     "document",
     "video",
diff --git a/vlmrun/version.py b/vlmrun/version.py
index 43a1e95..6b27eee 100644
--- a/vlmrun/version.py
+++ b/vlmrun/version.py
@@ -1 +1 @@
-__version__ = "0.5.3"
+__version__ = "0.5.4"