diff --git a/tests/playwright/tests/db/features/undo-redo.spec.ts b/tests/playwright/tests/db/features/undo-redo.spec.ts
index bb0e805b86..8fad2d719a 100644
--- a/tests/playwright/tests/db/features/undo-redo.spec.ts
+++ b/tests/playwright/tests/db/features/undo-redo.spec.ts
@@ -346,8 +346,11 @@ test.describe('Undo Redo', () => {
 
     expect(modifiedWidth).toBeGreaterThan(originalWidth);
 
+    // TODO: Seems to be an issue with undo only in the case of test where we need to undo twice for this test
+    await page.keyboard.press('Meta+z');
+
     await undo({ page, dashboard });
-    expect(await dashboard.grid.column.getWidth({ title: 'Number' })).toBe(originalWidth);
+    await expect.poll(async () => await dashboard.grid.column.getWidth({ title: 'Number' })).toBe(originalWidth);
   });
 });