diff --git a/lib/toc_data.rb b/lib/toc_data.rb index 841ae87..984d811 100644 --- a/lib/toc_data.rb +++ b/lib/toc_data.rb @@ -8,7 +8,7 @@ def toc_data(page_content) html_doc.css('h1, h2, h3').each do |header| headers.push({ id: header.attribute('id').to_s, - content: header.content, + content: header.children, level: header.name[1].to_i, children: [] })