diff --git a/js/packages/teams-ai/src/augmentations/SequenceAugmentation.spec.ts b/js/packages/teams-ai/src/augmentations/SequenceAugmentation.spec.ts
index fc3950f37..39e5a9f34 100644
--- a/js/packages/teams-ai/src/augmentations/SequenceAugmentation.spec.ts
+++ b/js/packages/teams-ai/src/augmentations/SequenceAugmentation.spec.ts
@@ -238,7 +238,19 @@ describe('SequenceAugmentation', () => {
                 // Create plan from response
                 const plan = await augmentation.createPlanFromResponse(context, state, {
                     status: 'success',
-                    message: { role: 'assistant', content: validation.value }
+                    message: {
+                        role: 'assistant',
+                        content: validation.value,
+                        context: {
+                            intent: 'to test citations',
+                            citations: [{
+                                url: '',
+                                title: '',
+                                filepath: '',
+                                content: 'my citation content...'
+                            }]
+                        }
+                    }
                 });
                 assert.notEqual(plan, undefined);
                 assert.deepEqual(plan, {
@@ -255,7 +267,16 @@ describe('SequenceAugmentation', () => {
                             type: 'SAY',
                             response: {
                                 role: 'assistant',
-                                content: 'hello world'
+                                content: 'hello world',
+                                context: {
+                                    intent: 'to test citations',
+                                    citations: [{
+                                        url: '',
+                                        title: '',
+                                        filepath: '',
+                                        content: 'my citation content...'
+                                    }]
+                                }
                             }
                         }
                     ]
diff --git a/js/packages/teams-ai/src/augmentations/SequenceAugmentation.ts b/js/packages/teams-ai/src/augmentations/SequenceAugmentation.ts
index 5f1e04b5a..64623bdd7 100644
--- a/js/packages/teams-ai/src/augmentations/SequenceAugmentation.ts
+++ b/js/packages/teams-ai/src/augmentations/SequenceAugmentation.ts
@@ -190,6 +190,7 @@ export class SequenceAugmentation implements Augmentation<Plan | undefined> {
                 return {
                     ...c,
                     response: {
+                        ...response.message,
                         role: 'assistant',
                         content: (c as PredictedSayCommand).response as unknown as string
                     }