diff --git a/src/md_to_html.py b/src/md_to_html.py
index 4cd10c7..3c5f0bc 100644
--- a/src/md_to_html.py
+++ b/src/md_to_html.py
@@ -347,14 +347,14 @@ def badge_findings_table(html_text):
                 cell_html,
             )
             cell_html = re.sub(
-                rf"(<h4>)\[{severity}\]",
+                rf"(<h4[^>]*>)\[{severity}\]",
                 rf'\1<span class="badge {css_class}">{severity}</span> ',
                 cell_html,
             )
         return cell_html
 
     html_text = re.sub(r"<td>\w+</td>", replace_cell, html_text)
-    html_text = re.sub(r"<h4>\[\w+\][^<]*", replace_cell, html_text)
+    html_text = re.sub(r"<h4[^>]*>\[\w+\][^<]*", replace_cell, html_text)
     return html_text