diff --git a/pywb/apps/rewriterapp.py b/pywb/apps/rewriterapp.py
index 442747b1..b8d7df7a 100644
--- a/pywb/apps/rewriterapp.py
+++ b/pywb/apps/rewriterapp.py
@@ -936,7 +936,7 @@ class RewriterApp(object):
                                                         self.frame_mod,
                                                         self.replay_mod,
                                                         self.client_side_replay,
-                                                        coll='',
+                                                        coll=kwargs.get("coll"),
                                                         extra_params=extra_params)
 
         return None
diff --git a/pywb/rewrite/templateview.py b/pywb/rewrite/templateview.py
index 07b867f3..a3ad3c8a 100644
--- a/pywb/rewrite/templateview.py
+++ b/pywb/rewrite/templateview.py
@@ -427,7 +427,9 @@ class TopFrameView(BaseInsertView):
                   'is_proxy': is_proxy,
                   'client_side_replay': client_side_replay,
                   'timestamp': timestamp,
-                  'url': wb_url.get_url()
+                  'url': wb_url.get_url(),
+
+                  'sw_prefix': env.get('pywb.app_prefix', '') + "/"
                  }
 
         if extra_params:
diff --git a/pywb/static/loadWabac.js b/pywb/static/loadWabac.js
index 400f6f0f..389aa395 100644
--- a/pywb/static/loadWabac.js
+++ b/pywb/static/loadWabac.js
@@ -1,18 +1,24 @@
 class WabacReplay
 {
-  constructor(prefix, url, ts, staticPrefix) {
+  constructor(prefix, url, ts, staticPrefix, coll, swScopePrefix) {
     this.prefix = prefix;
     this.url = url;
     this.ts = ts;
     this.staticPrefix = staticPrefix;
-    this.collName = new URL(prefix, "http://dummy").pathname.split('/')[1];
+    this.collName = coll;
+    this.isRoot = coll === "$root";
+    this.archivePrefix = this.isRoot ? "/" : `/${this.collName}/`;
+    this.swScope = swScopePrefix;
     this.adblockUrl = undefined;
 
     this.queryParams = {};
+    if (this.isRoot) {
+      this.queryParams["root"] = "$root";
+    }
   }
 
   async init() {
-    const scope = "/";
+    const scope = this.swScope;
 
     await navigator.serviceWorker.register(
       `${this.staticPrefix}/sw.js?` + new URLSearchParams(this.queryParams).toString(),
@@ -30,24 +36,22 @@ class WabacReplay
       }
     });
 
-    const baseUrl = new URL(window.location);
-    baseUrl.hash = "";
-
     const proxyPrefix = "";
 
     const msg = {
       msg_type: "addColl",
       name: this.collName,
       type: "live",
+      root: this.isRoot,
       file: {"sourceUrl": `proxy:${proxyPrefix}`},
       skipExisting: true,
       extraConfig: {
         prefix: proxyPrefix,
         isLive: false,
-        baseUrl: baseUrl.href,
-        baseUrlHashReplay: true,
+        baseUrl: this.prefix,
+        baseUrlAppendReplay: true,
         noPostToGet: false,
-        archivePrefix: `/${this.collName}/`,
+        archivePrefix: this.archivePrefix,
         archiveMod: "ir_",
         adblockUrl: this.adblockUrl
       },
@@ -80,6 +84,6 @@ class WabacReplay
   // called by the Vue banner when the timeline is clicked
   load_url(url, ts) {
     const iframe = document.querySelector('#replay_iframe');
-    iframe.src = `/w/${this.collName}/${ts}mp_/${url}`;
+    iframe.src = `${this.swScope}w${this.archivePrefix}${ts}mp_/${url}`;
   }
 }
diff --git a/pywb/templates/frame_insert.html b/pywb/templates/frame_insert.html
index bdb0b2ee..5e363e11 100644
--- a/pywb/templates/frame_insert.html
+++ b/pywb/templates/frame_insert.html
@@ -16,7 +16,7 @@ html, body
 {% if client_side_replay %}
 <script src='{{ static_prefix }}/loadWabac.js'></script>
 <script>
-  new WabacReplay("{{ wb_prefix }}", "{{ url }}", "{{ timestamp }}", "{{ static_prefix }}").init();
+  new WabacReplay("{{ wb_prefix }}", "{{ url }}", "{{ timestamp }}", "{{ static_prefix }}", "{{ coll }}", "{{ sw_prefix }}").init();
 </script>
 {% else %}
 <script src='{{ static_prefix }}/wb_frame.js'> </script>