diff --git a/app/src/protyle/render/av/layout.ts b/app/src/protyle/render/av/layout.ts
index 6dd974b10..3e8df8d59 100644
--- a/app/src/protyle/render/av/layout.ts
+++ b/app/src/protyle/render/av/layout.ts
@@ -179,7 +179,7 @@ export const bindLayoutEvent = (options: {
 };
 
 export const updateLayout = (options: {
-    view: IAVGallery
+    data: IAV
     nodeElement: Element,
     protyle: IProtyle,
     target: HTMLElement
@@ -199,9 +199,8 @@ export const updateLayout = (options: {
         action: "changeAttrViewLayout",
         avID,
         blockID,
-        layout: options.view.type
+        layout: options.data.viewType
     }]);
-    options.nodeElement.setAttribute("data-view-type", layout);
     options.target.parentElement.querySelector(".av__layout-item--select").classList.remove("av__layout-item--select");
     options.target.classList.add("av__layout-item--select");
 };
diff --git a/app/src/protyle/render/av/openMenuPanel.ts b/app/src/protyle/render/av/openMenuPanel.ts
index 55cf82749..9307e23e9 100644
--- a/app/src/protyle/render/av/openMenuPanel.ts
+++ b/app/src/protyle/render/av/openMenuPanel.ts
@@ -1386,7 +1386,7 @@ export const openMenuPanel = (options: {
                         target,
                         protyle: options.protyle,
                         nodeElement: options.blockElement,
-                        view: data.view as IAVGallery
+                        data: data
                     });
                     event.preventDefault();
                     event.stopPropagation();
diff --git a/app/src/protyle/render/av/render.ts b/app/src/protyle/render/av/render.ts
index 786492d11..5e7b5854c 100644
--- a/app/src/protyle/render/av/render.ts
+++ b/app/src/protyle/render/av/render.ts
@@ -488,6 +488,9 @@ export const refreshAV = (protyle: IProtyle, operation: IOperation) => {
                 if (operation.action === "removeAttrViewView") {
                     item.setAttribute("data-av-type", operation.retData);
                 }
+                if (operation.action === "changeAttrViewLayout") {
+                    item.setAttribute("data-av-type", operation.layout);
+                }
                 avRender(item, protyle, () => {
                     const attrElement = document.querySelector(`.b3-dialog--open[data-key="${Constants.DIALOG_ATTR}"] div[data-av-id="${avID}"]`) as HTMLElement;
                     if (attrElement) {