diff --git a/app/src/protyle/render/av/col.ts b/app/src/protyle/render/av/col.ts
index 5f4bf88ba..8d07859d5 100644
--- a/app/src/protyle/render/av/col.ts
+++ b/app/src/protyle/render/av/col.ts
@@ -937,6 +937,7 @@ export const showColMenu = (protyle: IProtyle, blockElement: Element, cellElemen
         bind(element) {
             const wrapElement = element.querySelector(".b3-switch") as HTMLInputElement;
             wrapElement.addEventListener("change", () => {
+                cellElement.dataset.wrap = wrapElement.checked.toString();
                 transaction(protyle, [{
                     action: "setAttrViewColWrap",
                     id: colId,