Fixed h1 with only an image being cleared

pgaskin · Jan 25, 2018 · 26fc1d2 · 26fc1d2
1 parent a4dea55
commit 26fc1d2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kepub/content.go b/kepub/content.go
@@ -235,7 +235,8 @@ func cleanHTML(doc *goquery.Document) error {
 
 	// Remove empty headings
 	doc.Find(`h1,h2,h3,h4,h5,h6`).FilterFunction(func(_ int, s *goquery.Selection) bool {
-		return strings.Trim(s.Text(), "\t \n") == ""
+		h, _ := s.Html()
+		return strings.Trim(h, "\t \n") == ""
 	}).Remove()
 
 	// Remove MS <st1:whatever> tags