diff --git a/layouts/_default/_markup/render-codeblock-mermaid.html b/layouts/_default/_markup/render-codeblock-mermaid.html
index ee8fb37464..e58431185b 100644
--- a/layouts/_default/_markup/render-codeblock-mermaid.html
+++ b/layouts/_default/_markup/render-codeblock-mermaid.html
@@ -1,4 +1,4 @@
-<div class="mermaid" align="center">
-  {{- safeHTML .Inner -}}
-</div>
-{{- .Page.Store.Set "htmlHasMermaid" true }}
+{{- partial "shortcodes/mermaid.html" (dict
+  "context" .
+  "content" .Inner
+) }}
\ No newline at end of file