diff --git a/package.json b/package.json
index 8304244..fc0fdf7 100644
--- a/package.json
+++ b/package.json
@@ -2,7 +2,7 @@
   "name": "antd-design-token",
   "displayName": "antd Design Token",
   "description": "VSCode extension for antd v5 design token.",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "publisher": "shezhangzhang",
   "engines": {
     "vscode": "^1.68.0"
diff --git a/src/listener.ts b/src/listener.ts
index e0b0534..3afedbf 100644
--- a/src/listener.ts
+++ b/src/listener.ts
@@ -23,22 +23,29 @@ export default function setupEventListener(
       }
 
       if (activeEditor && event.document === activeEditor.document) {
-        diffLine = activeEditor.document.lineCount - fileLineCount;
-        fileLineCount = activeEditor.document.lineCount;
+        /**
+         * redo and undo
+         */
+        if (event.reason) {
+          decorationManager.triggerUpdateDecorations(true);
+        } else {
+          diffLine = activeEditor.document.lineCount - fileLineCount;
+          fileLineCount = activeEditor.document.lineCount;
 
-        const [startLine, endLine, originalStartLine, originalEndLine] =
-          getStartEndLine(event.document, event.contentChanges[0]);
+          const [startLine, endLine] = getStartEndLine(
+            event.document,
+            event.contentChanges[0]
+          );
 
-        decorationManager.setActiveEditor(activeEditor);
-        decorationManager.triggerUpdateDecorations(
-          event.reason ? false : diffLine === 0,
-          true,
-          diffLine,
-          startLine,
-          endLine,
-          originalStartLine,
-          originalEndLine
-        );
+          decorationManager.setActiveEditor(activeEditor);
+          decorationManager.triggerUpdateDecorations(
+            diffLine === 0,
+            true,
+            diffLine,
+            startLine,
+            endLine
+          );
+        }
       }
     },
     null,