diff --git a/app/src/layout/topBar.ts b/app/src/layout/topBar.ts
index 5a44b5d5e..d3c3c1c8c 100644
--- a/app/src/layout/topBar.ts
+++ b/app/src/layout/topBar.ts
@@ -15,7 +15,7 @@ import {webFrame} from "electron";
 /// #endif
 import {Constants} from "../constants";
 import {isBrowser, isWindow} from "../util/functions";
-import {Menu} from "../plugin/API";
+import {Menu} from "../plugin/Meun";
 
 export const updateEditModeElement = () => {
     const target = document.querySelector("#barReadonly");
diff --git a/app/src/protyle/render/av/render.ts b/app/src/protyle/render/av/render.ts
index 6f8345e5e..e182e6f3f 100644
--- a/app/src/protyle/render/av/render.ts
+++ b/app/src/protyle/render/av/render.ts
@@ -88,7 +88,7 @@ export const avRender = (element: Element, cb?: () => void) => {
 
 export const refreshAV = (protyle: IProtyle, operation: IOperation) => {
     if (operation.action === "addAttrViewCol") {
-        Array.from(protyle.wysiwyg.element.querySelectorAll(`[data-node-id="${operation.id}"]`)).forEach((item: HTMLElement) => {
+        Array.from(protyle.wysiwyg.element.querySelectorAll(`[data-av-id="${operation.parentID}"]`)).forEach((item: HTMLElement) => {
             item.removeAttribute("data-render");
             avRender(item, () => {
                 showHeaderCellMenu(protyle, item, item.querySelector(".av__row--header").lastElementChild.previousElementSibling as HTMLElement);