Minor bug fixes

manifest.json

@@ -1,7 +1,7 @@
 {
 	"id": "obsidian-embedded-note-titles",
 	"name": "Embedded Note Titles",
-	"version": "1.2.2",
+	"version": "1.2.3",
 	"minAppVersion": "0.13.8",
 	"description": "Inserts the note file name as an H1 heading above each note.",
 	"author": "mgmeyers",

src/main.ts

@@ -104,14 +104,20 @@ export default class EmbeddedNoteTitlesPlugin extends Plugin {
           this.previewHeadingsManager.createHeadings(this.app);
         }, 0);
 
-        setTimeout(() => {
-          this.observedTitles.forEach((_, el) => {
-            if (!document.getElementById(el.id)) {
-              this.unobserveTitle(el);
-              el.remove();
-            }
-          });
-        }, 100);
+        if (!this.isLegacyEditor) {
+          setTimeout(() => {
+            this.observedTitles.forEach((_, el) => {
+              if (
+                this.app.workspace
+                  .getLeavesOfType("markdown")
+                  .every((leaf) => !leaf.view.containerEl.find(`#${el.id}`))
+              ) {
+                this.unobserveTitle(el);
+                el.remove();
+              }
+            });
+          }, 100);
+        }
       })
     );
 
@@ -156,8 +162,10 @@ export default class EmbeddedNoteTitlesPlugin extends Plugin {
   }
 
   unobserveTitle(el: HTMLElement) {
-    this.observedTitles.delete(el);
-    this.observer.unobserve(el);
+    if (this.observedTitles.has(el)) {
+      this.observedTitles.delete(el);
+      this.observer.unobserve(el);
+    }
   }
 
   async loadSettings() {

src/titleDecoration.ts

@@ -110,6 +110,17 @@ export function buildTitleDecoration(
             view.state.field(editorViewField)
           );
 
+          // This shouldn't happen, but just to be safe, remove any straggling titles
+          view.contentDOM.parentElement.childNodes.forEach((node) => {
+            if (
+              node instanceof HTMLElement &&
+              node.hasClass("embedded-note-title")
+            ) {
+              plugin.unobserveTitle(node);
+              node.remove();
+            }
+          });
+
           this.header = createEl("h1", {
             text: this.title,
             cls: `cm-line embedded-note-title embedded-note-title__edit${