diff --git a/packages/app/src/composer/draft/create-flow.test.ts b/packages/app/src/composer/draft/create-flow.test.ts
index 8587fa3db6..20b53acba4 100644
--- a/packages/app/src/composer/draft/create-flow.test.ts
+++ b/packages/app/src/composer/draft/create-flow.test.ts
@@ -242,10 +242,10 @@ describe("useDraftAgentCreateFlow", () => {
     expect(createRequest).toHaveBeenCalledWith({
       attempt: expect.objectContaining({
         text: "",
-        attachments: [attachment.attachment],
+        attachments: [{ ...attachment.attachment, chatHistorySource: attachment.source }],
       }),
       text: "",
-      attachments: [attachment.attachment],
+      attachments: [{ ...attachment.attachment, chatHistorySource: attachment.source }],
       cwd: "/repo",
     });
     expect(onCreateSuccess).toHaveBeenCalledTimes(1);