1 files changed, 4 insertions, 1 deletions
diff --git a/packages/core/src/tools/modifiable-tool.test.ts b/packages/core/src/tools/modifiable-tool.test.ts
index 56c27fe0..850da02b 100644
--- a/packages/core/src/tools/modifiable-tool.test.ts
+++ b/packages/core/src/tools/modifiable-tool.test.ts
@@ -74,9 +74,10 @@ describe('modifyWithEditor', () => {
       getProposedContent: vi.fn().mockResolvedValue(proposedContent),
       createUpdatedParams: vi
         .fn()
-        .mockImplementation((modifiedContent, originalParams) => ({
+        .mockImplementation((oldContent, modifiedContent, originalParams) => ({
           ...originalParams,
           modifiedContent,
+          oldContent,
         })),
     };
 
@@ -153,6 +154,7 @@ describe('modifyWithEditor', () => {
       );
 
       expect(mockModifyContext.createUpdatedParams).toHaveBeenCalledWith(
+        currentContent,
         modifiedContent,
         mockParams,
       );
@@ -183,6 +185,7 @@ describe('modifyWithEditor', () => {
         updatedParams: {
           ...mockParams,
           modifiedContent,
+          oldContent: currentContent,
         },
         updatedDiff: 'mock diff content',
       });