From cada78ca4d08c27760b5acf3b19c2ab6561cf2ab Mon Sep 17 00:00:00 2001
From: "Veronica K. B. Olsen" <1619840+vkbo@users.noreply.github.com>
Date: Mon, 1 Feb 2021 23:19:24 +0100
Subject: [PATCH] use what was added in #637 to replace first title lookup in
 editor saveText

---
 nw/gui/doceditor.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/nw/gui/doceditor.py b/nw/gui/doceditor.py
index a6b6f3ed..e5930690 100644
--- a/nw/gui/doceditor.py
+++ b/nw/gui/doceditor.py
@@ -416,17 +416,22 @@ class GuiDocEditor(QTextEdit):
 
         self.theIndex.scanText(tHandle, docText)
 
-        hLevel, _ = self.theParent.theIndex.getFirstTitle(tHandle)
+        if self._updateHeaders(checkLevel=True):
+            if self.theParent.projTabs.currentIndex() == self.theParent.idxNovelView:
+                logger.verbose("Document headers have changed, updating novel tree")
+                self.theParent.novelView.refreshTree()
+        else:
+            self.theParent.novelView.updateWordCounts(tHandle)
+
+        hLevel = "H0"
+        if self.theHeaders:
+            hLevel = self.theHeaders[0][1]
+
         if self.theProject.projTree.updateItemLayout(tHandle, hLevel):
             self.theParent.treeView.setTreeItemValues(tHandle)
             self.nwDocument.saveDocument(docText)
             self.docFooter.updateInfo()
 
-        if self._updateHeaders(checkLevel=True):
-            self.theParent.novelView.refreshTree()
-        else:
-            self.theParent.novelView.updateWordCounts(tHandle)
-
         return True
 
     def updateDocMargins(self):