1 files changed, 18 insertions, 4 deletions
diff --git a/lib/output/modifiers/addHeadingId.js b/lib/output/modifiers/addHeadingId.js
index 751f4b8..e2e2720 100644
--- a/lib/output/modifiers/addHeadingId.js
+++ b/lib/output/modifiers/addHeadingId.js
@@ -1,9 +1,23 @@
 var slug = require('github-slugid');
-var HTMLModifier = require('./html');
+var editHTMLElement = require('./editHTMLElement');
 
-var addHeadingID = HTMLModifier('h1,h2,h3,h4,h5,h6', function(heading) {
+/**
+    Add ID to an heading
+
+    @param {HTMLElement} heading
+*/
+function addId(heading) {
     if (heading.attr('id')) return;
     heading.attr('id', slug(heading.text()));
-});
+}
+
+/**
+    Add ID to all headings
+
+    @param {HTMLDom} $
+*/
+function addHeadingId($) {
+    return editHTMLElement($, 'h1,h2,h3,h4,h5,h6', addId);
+}
 
-module.exports = addHeadingID;
+module.exports = addHeadingId;