diff --git a/__tests__/e2e.test.js b/__tests__/e2e.test.js
index 839f96e..5f14029 100644
--- a/__tests__/e2e.test.js
+++ b/__tests__/e2e.test.js
@@ -65,7 +65,7 @@ describe('Screenshot testing', function() {
       assert.ok(rateList.length > 0);
       rateList.forEach((data) => {
         // console.log(`${data.path}]diff-rate: ${data.rate}`, );
-        assert.ok(data.rate < 0.05);
+        assert.ok(data.rate < 0.005);
       });
 
       done();
diff --git a/packages/core/examples/features/lib/data/rect.js b/packages/core/examples/features/lib/data/rect.js
index 5ee357f..81ab37e 100644
--- a/packages/core/examples/features/lib/data/rect.js
+++ b/packages/core/examples/features/lib/data/rect.js
@@ -1,5 +1,5 @@
 const data = {
-  bgColor: '#f0f0f0',
+  // bgColor: '#f0f0f0',
   elements: [
     {
       name: "rect-001",
@@ -56,7 +56,7 @@ const data = {
       desc: {
         color: "#e0e0e0",
         borderRadius: 20,
-        borderWidth: 2,
+        borderWidth: 10,
         borderColor: "#bd0b64",
       },
     },
diff --git a/packages/idraw/__tests__/index.test.ts b/packages/idraw/__tests__/index.test.ts
index 4cff47f..6cec08c 100644
--- a/packages/idraw/__tests__/index.test.ts
+++ b/packages/idraw/__tests__/index.test.ts
@@ -69,14 +69,23 @@ describe("idraw", () => {
     await delay(10);
 
     const undo1 = idraw.undo();
-    expect(undo1).toBe(2);
+    expect(undo1.doRecordCount).toBe(2);
+    expect(undo1.data?.elements?.length).toBe(4);
+    
+
     const undo2 = idraw.undo();
-    expect(undo2).toBe(1);
+    expect(undo2.doRecordCount).toBe(1);
+    expect(undo2.data?.elements?.length).toBe(4);
+
     const redo1 = idraw.redo();
-    expect(redo1).toBe(1);
+    expect(redo1.undoRecordCount).toBe(1);
+    expect(redo1.data?.elements?.length).toBe(4);
+
     idraw.moveDownElement('image-003');
+
     const redo2 = idraw.redo();
-    expect(redo2).toBe(0);
+    expect(redo2.undoRecordCount).toBe(0);
+    expect(redo2.data).toBe(null);
 
     requestAnimationFrameMock.triggerNextAnimationFrame();