diff --git a/lib/main/build-site.js b/lib/main/build-site.js index 15b57a2b7..f3b32b478 100644 --- a/lib/main/build-site.js +++ b/lib/main/build-site.js @@ -15,14 +15,13 @@ function build(type) { $header = $('h1').first().remove(), version = _.trim($header.find('span').first().text()); - $('h4').each(function() { - $(this) - .attr('class', null) - .attr('id', null) - .find('a').each(function() { - var $this = $(this); - $this.replaceWith($this.text()); - }); + $('[id^="user-content-"]') + .attr('class', null) + .attr('id', null); + + $(':header > a').each(function() { + var $a = $(this); + $a.replaceWith($a.html()); }); var html = $.html().replace(/(<)!--\s*|\s*--(>)/g, '$1$2');