From da3702352c4b59951c440e47ecd330117c7aa4e7 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Sat, 9 Nov 2024 14:19:23 +1100
Subject: [PATCH] Fix undo-redo expectations

---
 browser_tests/changeTracker.spec.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/browser_tests/changeTracker.spec.ts b/browser_tests/changeTracker.spec.ts
index 995e2cea5..26dce70a0 100644
--- a/browser_tests/changeTracker.spec.ts
+++ b/browser_tests/changeTracker.spec.ts
@@ -64,7 +64,7 @@ test.describe('Change Tracker', () => {
     await expect(comfyPage.canvas).toHaveScreenshot('undo-step1.png')
 
     comfyPage.ctrlZ()
-    await expect(comfyPage.canvas).toHaveScreenshot('undo-step2.png')
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-initial-state.png')
 
     comfyPage.dragAndDrop(
       comfyPage.clipTextEncodeNode1OutputSlot,
@@ -73,7 +73,10 @@ test.describe('Change Tracker', () => {
     await expect(comfyPage.canvas).toHaveScreenshot('undo-step3.png')
 
     comfyPage.ctrlZ()
-    await expect(comfyPage.canvas).toHaveScreenshot('undo-step4.png')
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-initial-state.png')
+
+    comfyPage.ctrlY()
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-step3.png')
   })
 
   test('Can group multiple change actions into a single transaction', async ({