diff --git a/server/embed/run.html b/server/embed/run.html
index c24f9274e..af94057a0 100644
--- a/server/embed/run.html
+++ b/server/embed/run.html
@@ -12,10 +12,10 @@
 
 <body>
   <h1></h1>
-  <div id="updateui" hidden>
+  <div id="updateprompt" hidden>
     Update available. Click <a href="#" onclick="location.reload()">here</a> to reload.
   </div>
-  <script type="module" src="/run" main="/main.mjs" sw="/embed/sw.js" updateui="#updateui"></script>
+  <script type="module" src="/run" main="/main.mjs" sw="/embed/sw.js" updateprompt="#updateprompt"></script>
 </body>
 
 </html>
diff --git a/server/embed/run.ts b/server/embed/run.ts
index 95f7624b2..50f87163e 100644
--- a/server/embed/run.ts
+++ b/server/embed/run.ts
@@ -6,8 +6,6 @@ import type { RunOptions } from "./types/run.d.ts";
 
 const document: Document | undefined = window.document;
 const kRun = "esm.sh/run";
-const kImportmap = "importmap";
-const localhosts = ["localhost", "127.0.0.1"];
 
 function run(options: RunOptions = {}): Promise<ServiceWorker> {
   const serviceWorker = navigator.serviceWorker;
@@ -17,27 +15,20 @@ function run(options: RunOptions = {}): Promise<ServiceWorker> {
   const hasController = serviceWorker.controller !== null;
   const onUpdateFound = options.onUpdateFound ?? (() => location.reload());
   return new Promise<ServiceWorker>(async (resolve, reject) => {
-    let sw = options.sw;
-    if (options.devSW && localhosts.includes(location.hostname)) {
-      sw = options.devSW;
-    }
-    const reg = await serviceWorker.register(sw ?? "/sw.js", {
+    const reg = await serviceWorker.register(options.sw ?? "/sw.js", {
       type: "module",
       scope: options.swScope,
     });
     const run = async () => {
       if (reg.active?.state === "activated") {
-        const importMapSupported = HTMLScriptElement.supports?.(kImportmap);
-        const imports: Record<string, string> = {};
-        const scopes: Record<string, typeof imports> = {};
         let p: Promise<boolean> | undefined;
+        let importMap: Record<string, any> | null = null;
         queryElement<HTMLScriptElement>('script[type="importmap"]', (el) => {
           try {
             const json = JSON.parse(el.textContent!);
-            for (const scope in json.scopes) {
-              scopes[scope] = { ...imports, ...json.imports };
+            if (json && typeof json === "object") {
+              importMap = json;
             }
-            Object.assign(imports, json.imports);
           } catch (e) {
             console.error("Failed to parse importmap:", e);
           }
@@ -126,14 +117,21 @@ queryElement<HTMLScriptElement>("script[type='module'][src][main]", (el) => {
   const src = el.src;
   const main = attr(el, "main");
   if (src === import.meta.url && main) {
-    const options: RunOptions = { main, sw: attr(el, "sw"), devSW: attr(el, "dev-sw") };
-    const updateui = attr(el, "updateui");
-    if (updateui) {
-      const el = document!.querySelector<HTMLElement>(updateui);
-      if (!el) {
-        throw new Error("Could not find the `updateui` element");
-      }
-      options.onUpdateFound = () => el.style.display = "block";
+    const options: RunOptions = { main, sw: attr(el, "sw") };
+    const updateprompt = attr(el, "updateprompt");
+    if (updateprompt) {
+      queryElement<HTMLElement>(updateprompt, (el) => {
+        options.onUpdateFound = () => {
+          if (el instanceof HTMLDialogElement) {
+            el.showModal();
+          } else {
+            el.hidden = false;
+            if (el.hasAttribute("popover")) {
+              el.showPopover?.();
+            }
+          }
+        };
+      });
     }
     run(options);
   }
diff --git a/server/embed/types/run.d.ts b/server/embed/types/run.d.ts
index 1720da4e3..2cb0e713b 100644
--- a/server/embed/types/run.d.ts
+++ b/server/embed/types/run.d.ts
@@ -9,7 +9,6 @@ export interface VFile {
 
 export interface RunOptions {
   main?: string;
-  devSW?: string | null;
   sw?: string | null;
   swScope?: string;
   onUpdateFound?: () => void;
