diff --git a/src/components/_shared/TableOfContents/TableOfContents.tsx b/src/components/_shared/TableOfContents/TableOfContents.tsx index 4320c30c..3b132709 100644 --- a/src/components/_shared/TableOfContents/TableOfContents.tsx +++ b/src/components/_shared/TableOfContents/TableOfContents.tsx @@ -43,14 +43,12 @@ export const TableOfContents = (props: Props) => { setIsVisible(isDesktop()) const updateHeadings = () => { - if (document) { - const parent = document.querySelector(props.parentSelector) - if (parent) { - setHeadings( - // eslint-disable-next-line unicorn/prefer-spread - Array.from(parent.querySelectorAll('h1, h2, h3, h4')) - ) - } + const parent = document?.querySelector(props.parentSelector) + if (parent) { + setHeadings( + // eslint-disable-next-line unicorn/prefer-spread + Array.from(parent.querySelectorAll('h1, h2, h3, h4')) + ) } setAreHeadingsLoaded(true) }