From df4d3ca4939802045f2a0dd7157fee44077c4ee6 Mon Sep 17 00:00:00 2001
From: Jackson Chen <jacksonc@spotify.com>
Date: Thu, 27 Jun 2024 15:33:39 -0400
Subject: [PATCH] Fix tests

Signed-off-by: Jackson Chen <jacksonc@spotify.com>
---
 plugins/techdocs-react/src/hooks.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/plugins/techdocs-react/src/hooks.ts b/plugins/techdocs-react/src/hooks.ts
index bbff875f3a..2d922ab7f4 100644
--- a/plugins/techdocs-react/src/hooks.ts
+++ b/plugins/techdocs-react/src/hooks.ts
@@ -39,13 +39,13 @@ export const useShadowRootElements = <
   selectors: string[],
 ): TReturnedElement[] => {
   const shadowRoot = useShadowRoot();
-  const [root, setRootNode] = useState(shadowRoot?.querySelector('html'));
+  const [render, rerender] = useState(false);
 
   useEffect(() => {
     let observer: MutationObserver;
     if (shadowRoot) {
       observer = new MutationObserver(() => {
-        setRootNode(shadowRoot.querySelector('html'));
+        rerender(!render);
       });
       observer.observe(shadowRoot, {
         attributes: true,
@@ -55,12 +55,12 @@ export const useShadowRootElements = <
       });
     }
     return () => observer?.disconnect();
-  }, [shadowRoot]);
+  }, [shadowRoot, render, rerender]);
 
-  if (!root) return [];
+  if (!shadowRoot) return [];
 
   return selectors
-    .map(selector => root.querySelectorAll<TReturnedElement>(selector))
+    .map(selector => shadowRoot.querySelectorAll<TReturnedElement>(selector))
     .filter(nodeList => nodeList.length)
     .map(nodeList => Array.from(nodeList))
     .flat();