diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py
index 52d732f7..7fecd511 100644
--- a/vision_agent/tools/meta_tools.py
+++ b/vision_agent/tools/meta_tools.py
@@ -676,7 +676,7 @@ def use_object_detection_fine_tuning(
     display(
         {
             MimeType.APPLICATION_ARTIFACT: json.dumps(
-                {"name": name, "content": new_code}
+                {"name": name, "content": new_code, "action": "edit"}
             )
         },
         raw=True,