diff --git a/index.js b/index.js index e031815..452332f 100644 --- a/index.js +++ b/index.js @@ -30,5 +30,5 @@ module.exports = (dir, fileTypes = null, isHtml = false) => { const archyTree = dirTreeToArchyTree(tree, dir, isHtml); const outTree = archy(archyTree); - return isHtml ? wrapHtml(outTree) : outTree; + return isHtml ? wrapHtml(outTree, tree.name) : outTree; } diff --git a/test/fixtures/1.html b/test/fixtures/1.html index 15a29f6..0fdf744 100644 --- a/test/fixtures/1.html +++ b/test/fixtures/1.html @@ -2,6 +2,7 @@ + 1