preserve h2 class when stickyalizing it

1 files changed, 8 insertions, 2 deletions

diff --git a/src/upd8.js b/src/upd8.js
index 920f903..a793feb 100755
--- a/src/upd8.js
+++ b/src/upd8.js
@@ -572,8 +572,14 @@ function transformMultiline(text, {
       }
 
       // for sticky headings!
-      if (elementMatch) {
-        lineContent = lineContent.replace(/<h2/, `<h2 class="content-heading"`)
+      if (elementMatch && elementMatch[1] === 'h2') {
+        lineContent = lineContent.replace(/<h2(.*?)>/g, (match, attributes) => {
+          const parsedAttributes = parseAttributes(attributes, {to});
+          return `<h2 ${html.attributes({
+            ...parsedAttributes,
+            class: [...parsedAttributes.class?.split(' ') ?? [], 'content-heading'],
+          })}>`;
+        });
       }
     }