Auto selecting first row after sort

[ishrest]

I have a sortable table that needs to auto select first row even after sorting. I am using v7 and was able to auto select first row on initial load by
initialState.selectedRowIds:{'0':true}

However, after sorting, the original row remains selected instead of the first row of sorted table.
I have tried adding selectedRowIds in useControlledState as below but that's not working.
useControlledState: state => { return useMemo( () => ({ ...state, selectedRowIds:{'0':true} }), [state] );

Any help will be greatly appreciated. Thanks in advance