diff --git a/server-rs/crates/api-server/src/external_generation_worker.rs b/server-rs/crates/api-server/src/external_generation_worker.rs
index 0f59ede8c..1f5e2ef53 100644
--- a/server-rs/crates/api-server/src/external_generation_worker.rs
+++ b/server-rs/crates/api-server/src/external_generation_worker.rs
@@ -1434,7 +1434,7 @@ mod tests {
     fn editor_agent_result_payload_keeps_compact_response() {
         let mut job = external_generation_job_record_fixture(Some("lease-1"));
         job.request_payload_json = json!({
-            "generationInputs": { "source": "editor-agent-tool-call-result" },
+            "generationInputs": { "source": "editor-agent" },
         })
         .to_string();
         let response = json!({
@@ -1484,7 +1484,7 @@ mod tests {
     fn editor_agent_spritesheet_result_keeps_all_persisted_slices() {
         let mut job = external_generation_job_record_fixture(Some("lease-1"));
         job.request_payload_json = json!({
-            "generationInputs": { "source": "editor-agent-tool-call-result" },
+            "generationInputs": { "source": "editor-agent" },
         })
         .to_string();
         let response = json!({