From 87bdffb3fa82673794fafb359b9c7f969256ee89 Mon Sep 17 00:00:00 2001
From: Vanessa <lly219@gmail.com>
Date: Mon, 4 Mar 2024 14:39:28 +0800
Subject: [PATCH] :art: https://github.com/siyuan-note/siyuan/issues/10443

---
 app/src/protyle/render/av/render.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/app/src/protyle/render/av/render.ts b/app/src/protyle/render/av/render.ts
index f91ad3d58..1e05a6861 100644
--- a/app/src/protyle/render/av/render.ts
+++ b/app/src/protyle/render/av/render.ts
@@ -53,6 +53,7 @@ export const avRender = (element: Element, protyle: IProtyle, cb?: () => void, v
             if (typeof viewID === "string") {
                 newViewID = viewID;
                 fetchPost("/api/av/setDatabaseBlockView", {id: e.dataset.nodeId, viewID});
+                e.setAttribute("custom-sy-av-view", newViewID);
             } else if (typeof viewID === "undefined") {
                 newViewID = e.querySelector(".av__header .item--focus")?.getAttribute("data-id");
             }
@@ -61,7 +62,7 @@ export const avRender = (element: Element, protyle: IProtyle, cb?: () => void, v
                 created,
                 snapshot,
                 pageSize: parseInt(e.dataset.pageSize) || undefined,
-                viewID: newViewID
+                viewID: newViewID || e.getAttribute("custom-sy-av-view")
             }, (response) => {
                 const data = response.data.view as IAVTable;
                 if (!e.dataset.pageSize) {