Don't cleanup StyledShadow on unmount

theatre/studio/src/css.tsx

@@ -113,14 +113,24 @@ const ProvideStyledShadow: React.FC<{
 
   useLayoutEffect(() => {
     if (!template) return
-    const shadowRoot = (template.parentNode as HTMLElement).attachShadow({
+    const {parentNode} = template
-      mode: 'open',
+    if (!parentNode) return
-    })
+
+    const hadShadowRoot =
+      // @ts-ignore
+      !!parentNode.shadowRoot
+
+    const shadowRoot = hadShadowRoot
+      ? // @ts-ignore
+        parent.shadowRoot
+      : (parentNode as HTMLElement).attachShadow({
+          mode: 'open',
+        })
+
     setShadowRoot(shadowRoot)
 
-    return () => {
+    // no need to cleanup. The parent will be removed anyway if cleanup
-      template.parentNode?.removeChild(shadowRoot)
+    // is needed.
-    }
   }, [template])
 
   const [portalLayerRef, portalLayer] = useRefAndState<HTMLDivElement>(