diff --git a/src/components/useList/hooks/useList.ts b/src/components/useList/hooks/useList.ts index 2446bf5b0..11a99c9eb 100644 --- a/src/components/useList/hooks/useList.ts +++ b/src/components/useList/hooks/useList.ts @@ -50,15 +50,6 @@ export const useList = ({ withExpandedState, }); - const {visibleFlattenIds, idToFlattenIndex, rootIds} = useFlattenListItems({ - items, - /** - * By default controlled from list items declaration state - */ - expandedById: innerState.expandedById, - getItemId, - }); - const realState = React.useMemo(() => { if (controlledState) { return { @@ -70,6 +61,15 @@ export const useList = ({ return innerState; }, [controlledState, innerState]); + const {visibleFlattenIds, idToFlattenIndex, rootIds} = useFlattenListItems({ + items, + /** + * By default controlled from list items declaration state + */ + expandedById: realState.expandedById, + getItemId, + }); + return { state: realState, structure: {