diff --git a/webui/src/web-components/demo/sketch-timeline-viewport.demo.ts b/webui/src/web-components/demo/sketch-timeline-viewport.demo.ts
index 0e53586..cc4b519 100644
--- a/webui/src/web-components/demo/sketch-timeline-viewport.demo.ts
+++ b/webui/src/web-components/demo/sketch-timeline-viewport.demo.ts
@@ -100,14 +100,11 @@
 
     // Helper functions
     const setupScrollContainer = () => {
-      if (timeline.shadowRoot) {
-        const scrollContainer =
-          timeline.shadowRoot.querySelector("#scroll-container");
-        if (scrollContainer) {
-          timeline.scrollContainer = { value: scrollContainer };
-          console.log("Scroll container set up:", scrollContainer);
-          return true;
-        }
+      const scrollContainer = timeline.querySelector("#scroll-container");
+      if (scrollContainer) {
+        timeline.scrollContainer = { value: scrollContainer };
+        console.log("Scroll container set up:", scrollContainer);
+        return true;
       }
       return false;
     };
@@ -118,12 +115,10 @@
       }
 
       const observer = new MutationObserver(() => {
-        if (timeline.shadowRoot) {
-          observer.disconnect();
-          timeline.updateComplete.then(() => {
-            setupScrollContainer();
-          });
-        }
+        observer.disconnect();
+        timeline.updateComplete.then(() => {
+          setupScrollContainer();
+        });
       });
 
       observer.observe(timeline, { childList: true, subtree: true });