diff --git a/.dumi/theme/slots/PreviewerActions.tsx b/.dumi/theme/slots/PreviewerActions.tsx index ddf4a68ceb..6640a73a57 100644 --- a/.dumi/theme/slots/PreviewerActions.tsx +++ b/.dumi/theme/slots/PreviewerActions.tsx @@ -202,15 +202,13 @@ const PreviewerActions: typeof DumiPreviewerActions = (props) => { useEffect(() => { if (div.current) { - const elements = div.current.querySelectorAll('.dumi-default-previewer-actions'); - elements.forEach((element) => { - element.addEventListener('click', (e) => { - e.stopImmediatePropagation(); - setLoading(true); - openCodeSandbox(props).finally(() => { - setLoading(false); - }); - }) + const element = div.current.querySelector('.dumi-default-previewer-action-btn'); + element?.addEventListener('click', (e) => { + e.stopImmediatePropagation(); + setLoading(true); + openCodeSandbox(props).finally(() => { + setLoading(false); + }); }) } }, [div]) diff --git a/packages/core/client/.dumi/theme/slots/PreviewerActions.tsx b/packages/core/client/.dumi/theme/slots/PreviewerActions.tsx index ddf4a68ceb..6640a73a57 100644 --- a/packages/core/client/.dumi/theme/slots/PreviewerActions.tsx +++ b/packages/core/client/.dumi/theme/slots/PreviewerActions.tsx @@ -202,15 +202,13 @@ const PreviewerActions: typeof DumiPreviewerActions = (props) => { useEffect(() => { if (div.current) { - const elements = div.current.querySelectorAll('.dumi-default-previewer-actions'); - elements.forEach((element) => { - element.addEventListener('click', (e) => { - e.stopImmediatePropagation(); - setLoading(true); - openCodeSandbox(props).finally(() => { - setLoading(false); - }); - }) + const element = div.current.querySelector('.dumi-default-previewer-action-btn'); + element?.addEventListener('click', (e) => { + e.stopImmediatePropagation(); + setLoading(true); + openCodeSandbox(props).finally(() => { + setLoading(false); + }); }) } }, [div])