diff --git a/src/components/image-editor/useImageCanvasGenerationSubmissionWorkflow.test.tsx b/src/components/image-editor/useImageCanvasGenerationSubmissionWorkflow.test.tsx
index a5554b8fb..282077556 100644
--- a/src/components/image-editor/useImageCanvasGenerationSubmissionWorkflow.test.tsx
+++ b/src/components/image-editor/useImageCanvasGenerationSubmissionWorkflow.test.tsx
@@ -2002,8 +2002,11 @@ describe('useImageCanvasGenerationSubmissionWorkflow', () => {
       makeInstrumental: true,
       projectId: 'editor-project-bgm',
       generationInputs: {
+        version: 2,
+        action: 'audio.background-music.generate',
         fields: [
           {
+            id: 'prompt',
             title: 'gpt_description_prompt',
             value: canonicalPrompt,
           },
@@ -2058,8 +2061,11 @@ describe('useImageCanvasGenerationSubmissionWorkflow', () => {
     expect(generatedLayer?.prompt).toBe(canonicalPrompt);
     expect(generatedLayer?.actualPrompt).toBe(canonicalPrompt);
     expect(generatedLayer?.generationInputs).toEqual({
+      version: 2,
+      action: 'audio.background-music.generate',
       fields: [
         {
+          id: 'prompt',
           title: 'gpt_description_prompt',
           value: canonicalPrompt,
         },