diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
index bfd2697b..38857724 100644
--- a/vision_agent/agent/vision_agent.py
+++ b/vision_agent/agent/vision_agent.py
@@ -471,7 +471,7 @@ def chat_with_artifacts(
                     self.streaming_message(
                         {
                             "role": "assistant",
-                            "content": json.dumps(response),
+                            "content": json.dumps(add_step_descriptions(response)),
                             "finished": finished and code_action is None,
                         }
                     )