diff --git a/packages/vscode-ide-companion/src/webview/hooks/useMessageSubmit.test.ts b/packages/vscode-ide-companion/src/webview/hooks/useMessageSubmit.test.ts
index 65d486f99..dd0479dbb 100644
--- a/packages/vscode-ide-companion/src/webview/hooks/useMessageSubmit.test.ts
+++ b/packages/vscode-ide-companion/src/webview/hooks/useMessageSubmit.test.ts
@@ -96,7 +96,16 @@ describe('shouldSendMessage', () => {
       shouldSendMessage({
         ...defaults,
         inputText: '',
-        attachedImages: [{ data: 'base64data', mediaType: 'image/png' }],
+        attachedImages: [
+          {
+            id: '1',
+            name: 'test.png',
+            type: 'image/png',
+            size: 100,
+            data: 'base64data',
+            timestamp: Date.now(),
+          },
+        ],
       }),
     ).toBe(true);
   });
@@ -106,7 +115,16 @@ describe('shouldSendMessage', () => {
       shouldSendMessage({
         ...defaults,
         inputText: '\u200B',
-        attachedImages: [{ data: 'base64data', mediaType: 'image/png' }],
+        attachedImages: [
+          {
+            id: '1',
+            name: 'test.png',
+            type: 'image/png',
+            size: 100,
+            data: 'base64data',
+            timestamp: Date.now(),
+          },
+        ],
       }),
     ).toBe(true);
   });