diff --git a/src/plugin.ts b/src/plugin.ts
index 3df0384..fea722c 100644
--- a/src/plugin.ts
+++ b/src/plugin.ts
@@ -51,31 +51,26 @@ body.dark, :root[data-theme="dark"] {
  */
 function createScript(version: string): string {
   return String.raw`
-<script src="https://unpkg.com/mermaid@${version}/dist/mermaid.min.js"></script>
-<script>
-(function() {
-  if (typeof mermaid === "undefined") {
-    return;
-  }
-
-  document.documentElement.classList.add("mermaid-enabled");
+<script type="module">
+import mermaid from "https://unpkg.com/mermaid@${version}/dist/mermaid.esm.min.mjs";
 
-  mermaid.initialize({startOnLoad:true});
+document.documentElement.classList.add("mermaid-enabled");
 
-  requestAnimationFrame(function check() {
-    let some = false;
-    document.querySelectorAll("div.mermaid:not([data-inserted])").forEach(div => {
-      some = true;
-      if (div.querySelector("svg")) {
-        div.dataset.inserted = true;
-      }
-    });
+mermaid.initialize({startOnLoad:true});
 
-    if (some) {
-      requestAnimationFrame(check);
+requestAnimationFrame(function check() {
+  let some = false;
+  document.querySelectorAll("div.mermaid:not([data-inserted])").forEach(div => {
+    some = true;
+    if (div.querySelector("svg")) {
+      div.dataset.inserted = true;
     }
   });
-})();
+
+  if (some) {
+    requestAnimationFrame(check);
+  }
+});
 </script>
 `;
 }
diff --git a/test/__snapshots__/plugin.test.ts.snap b/test/__snapshots__/plugin.test.ts.snap
index 06dce14..194ce66 100644
--- a/test/__snapshots__/plugin.test.ts.snap
+++ b/test/__snapshots__/plugin.test.ts.snap
@@ -66,31 +66,26 @@ body.dark, :root[data-theme=\\"dark\\"] {
 }
 </style>
 </head><body><div class=\\"mermaid-block\\"></div>
-<script src=\\"https://unpkg.com/mermaid@latest/dist/mermaid.min.js\\"></script>
-<script>
-(function() {
-  if (typeof mermaid === \\"undefined\\") {
-    return;
-  }
-
-  document.documentElement.classList.add(\\"mermaid-enabled\\");
+<script type=\\"module\\">
+import mermaid from \\"https://unpkg.com/mermaid@latest/dist/mermaid.esm.min.mjs\\";
 
-  mermaid.initialize({startOnLoad:true});
+document.documentElement.classList.add(\\"mermaid-enabled\\");
 
-  requestAnimationFrame(function check() {
-    let some = false;
-    document.querySelectorAll(\\"div.mermaid:not([data-inserted])\\").forEach(div => {
-      some = true;
-      if (div.querySelector(\\"svg\\")) {
-        div.dataset.inserted = true;
-      }
-    });
+mermaid.initialize({startOnLoad:true});
 
-    if (some) {
-      requestAnimationFrame(check);
+requestAnimationFrame(function check() {
+  let some = false;
+  document.querySelectorAll(\\"div.mermaid:not([data-inserted])\\").forEach(div => {
+    some = true;
+    if (div.querySelector(\\"svg\\")) {
+      div.dataset.inserted = true;
     }
   });
-})();
+
+  if (some) {
+    requestAnimationFrame(check);
+  }
+});
 </script>
 </body>"
 `;
diff --git a/test/plugin.test.ts b/test/plugin.test.ts
index 4872d22..4f95469 100644
--- a/test/plugin.test.ts
+++ b/test/plugin.test.ts
@@ -19,7 +19,7 @@ describe('MermaidPlugin', () => {
     const result = plugin.insertMermaidScript(input);
     expect(result).toMatch('</body>');
     expect(result).toMatch('mermaid.initialize({');
-    expect(result).toMatch(/src="https:\/\/unpkg.com\/mermaid\@latest\/dist\/mermaid.min.js"/);
+    expect(result).toMatch(/import mermaid from "https:\/\/unpkg.com\/mermaid\@latest\/dist\/mermaid.esm.min.mjs";/);
     expect(result).toMatchSnapshot();
   });
 
