diff --git a/app/src/protyle/toolbar/index.ts b/app/src/protyle/toolbar/index.ts
index 5519a7da2..31d2f04df 100644
--- a/app/src/protyle/toolbar/index.ts
+++ b/app/src/protyle/toolbar/index.ts
@@ -646,13 +646,12 @@ export class Toolbar {
                                 (currentNode.getAttribute("data-inline-memo-content") || ""));
                         }
                     }
+                    startContainer = currentNode;
                     if (currentType.includes("inline-math")) {
-                        startContainer = currentNode;
                         startOffset = null;
                     } else {
                         if (i === newNodes.length) {
                             endOffset = previousElement.textContent.length;
-                            startContainer = currentNode;
                         } else {
                             startOffset = previousElement.textContent.length;
                         }