Support non atx header

1 files changed, 3 insertions, 3 deletions

diff --git a/index.js b/index.js
index 8acc9f9..3ad0dc8 100644
--- a/index.js
+++ b/index.js
@@ -3,8 +3,8 @@
 const parseAttr = require('md-attr-parser');
 const htmlElemAttr = require('html-element-attributes');
 
-const supportedElements = ['link', 'atxHeading', 'strong', 'emphasis', 'deletion', 'code'];
-const blockElements = ['atxHeading'];
+const supportedElements = ['link', 'atxHeading', 'strong', 'emphasis', 'deletion', 'code', 'setextHeading'];
+const blockElements = ['atxHeading', 'setextHeading'];
 
 const DOMEventHandler = [
   'onabort', 'onautocomplete', 'onautocompleteerror',
@@ -165,7 +165,7 @@ function remarkAttr(userConfig) {
     if (supportedElements.indexOf(elem) >= 0) {
       if (blockElements.indexOf(elem) >= 0) {
         const oldElem = tokenizersBlock[elem];
-        tokenizersBlock.atxHeading = tokenizeGenerator('\n', oldElem, config);
+        tokenizersBlock[elem] = tokenizeGenerator('\n', oldElem, config);
       } else {
         const oldElem = tokenizers[elem];
         const elemTokenize = tokenizeGenerator('', oldElem, config);