diff --git a/chatlab/views/tools.py b/chatlab/views/tools.py
index da01f8a..87a7db1 100644
--- a/chatlab/views/tools.py
+++ b/chatlab/views/tools.py
@@ -110,7 +110,8 @@ def append_arguments(self, arguments: str):
     def apply_result(self, result: str):
         """Replaces the existing display with a new one that shows the result of the tool being called."""
         tc = ToolCalled(
-            id=self.id, name=self.name, arguments=self.arguments, result=result, display_id=self.display_id
+            id=self.id, name=self.name, arguments=self.arguments, result=result, display_id=self.display_id,
+            custom_render=self.custom_render
         )
         tc.update()
         return tc