diff --git a/src/WidgetApp.tsx b/src/WidgetApp.tsx index 043a7c1..89badfd 100644 --- a/src/WidgetApp.tsx +++ b/src/WidgetApp.tsx @@ -1,5 +1,5 @@ import { Box } from "@mui/material"; -import { startTransition, useEffect, useRef, useState } from "react"; +import { startTransition, useEffect, useLayoutEffect, useRef, useState } from "react"; import { QuizIdContext } from "@contexts/QuizIdContext"; import { RootContainerWidthContext } from "@contexts/RootContainerWidthContext"; import QuizAnswerer from "../lib/components/QuizAnswerer"; @@ -13,6 +13,10 @@ export default function WidgetApp({ quizId }: Props) { const [rootContainerSize, setRootContainerSize] = useState(() => window.innerWidth); const rootContainerRef = useRef(null); + useLayoutEffect(() => { + if (rootContainerRef.current) setRootContainerSize(rootContainerRef.current.clientWidth); + }, []); + useEffect(() => { const handleWindowResize = () => { startTransition(() => {