diff --git a/packages/x-data-grid/src/hooks/features/rows/useGridRowSpanning.ts b/packages/x-data-grid/src/hooks/features/rows/useGridRowSpanning.ts index ef9dafab23218..a588295e0e78f 100644 --- a/packages/x-data-grid/src/hooks/features/rows/useGridRowSpanning.ts +++ b/packages/x-data-grid/src/hooks/features/rows/useGridRowSpanning.ts @@ -3,7 +3,6 @@ import useLazyRef from '@mui/utils/useLazyRef'; import { GRID_DETAIL_PANEL_TOGGLE_FIELD } from '../../../internals/constants'; import { gridVisibleColumnDefinitionsSelector } from '../columns/gridColumnsSelector'; import { useGridVisibleRows } from '../../utils/useGridVisibleRows'; -import { gridRenderContextSelector } from '../virtualization/gridVirtualizationSelectors'; import { useGridSelector } from '../../utils/useGridSelector'; import type { GridColDef } from '../../../models/colDef'; import type { GridRowId, GridValidRowModel, GridRowEntry } from '../../../models/gridRows'; @@ -17,6 +16,7 @@ import { getCellValue, } from './gridRowSpanningUtils'; import { GRID_CHECKBOX_SELECTION_FIELD } from '../../../colDef/gridCheckboxSelectionColDef'; +import type { GridRenderContext } from '../../../models'; export interface GridRowSpanningState { spannedCells: Record>; @@ -224,7 +224,7 @@ export const useGridRowSpanning = ( props: Pick, ): void => { const { range, rows: visibleRows } = useGridVisibleRows(apiRef, props); - const renderContext = useGridSelector(apiRef, gridRenderContextSelector); + const renderContext = range ?? EMPTY_RANGE; const colDefs = useGridSelector(apiRef, gridVisibleColumnDefinitionsSelector); const processedRange = useLazyRef(() => { return Object.keys(apiRef.current.state.rowSpanning.spannedCells).length > 0 @@ -254,7 +254,7 @@ export const useGridRowSpanning = ( return; } - if (range === null || !isRowContextInitialized(renderContext)) { + if (range === null || !isRowContextInitialized(renderContext as GridRenderContext)) { return; }