diff --git a/src/common/useDeepEqualMemo.js b/src/common/useDeepEqualMemo.js index f3ad91680..e0242603f 100644 --- a/src/common/useDeepEqualMemo.js +++ b/src/common/useDeepEqualMemo.js @@ -4,12 +4,16 @@ const React = require('react'); const isEqual = require('lodash.isequal'); const useDeepEqualMemo = (cb, deps) => { - const valueRef = React.useRef(cb()); + const valueRef = React.useRef(); + const mountedRef = React.useRef(false); const prevDepsRef = React.useRef(deps); - if (!isEqual(prevDepsRef.current, deps)) { + if (!mountedRef.current || !isEqual(prevDepsRef.current, deps)) { valueRef.current = cb(); prevDepsRef.current = deps; } + React.useLayoutEffect(() => { + mountedRef.current = true; + }, []); return valueRef.current; };