diff --git a/package.json b/package.json
index 6829931..2d5fe08 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "markdown-it-textual-uml",
-  "version": "0.17.0",
+  "version": "0.17.1",
   "description": "Markdown-it markdown parser plugin to create uml diagrams from text, based on plantuml, mermaid, etc.",
   "main": "src/index.js",
   "type": "module",
diff --git a/src/mermaid-parser.js b/src/mermaid-parser.js
index 4726850..25a72ab 100644
--- a/src/mermaid-parser.js
+++ b/src/mermaid-parser.js
@@ -11,7 +11,7 @@ const functions = {
 
   getMarkup(code) {
     let content = removeTripleBackticks(code)
-    return `<pre><div class="mermaid">\n${content}\n</div></pre>\n`
+    return `<pre class="mermaid">\n${content}\n</pre>\n`
   },
 }