diff --git a/ui/effects/use-screensize.js b/ui/effects/use-screensize.js index b6ade3a91..0b2ea5b36 100644 --- a/ui/effects/use-screensize.js +++ b/ui/effects/use-screensize.js @@ -3,7 +3,8 @@ import React from 'react'; export function useWindowSize(fn) { const isWindowClient = typeof window === 'object'; - const [windowSize, setWindowSize] = React.useState(isWindowClient ? window.innerWidth : undefined); + const initialState = fn ? fn(window.innerWidth) : window.innerWidth; + const [windowSize, setWindowSize] = React.useState(isWindowClient ? initialState : undefined); React.useEffect(() => { function setSize() {