Maintain Row selection in SubComponent on re-render

[rokin04]

I'm rendering Table on row expansion , when I select any row in subcomponent and toggle the main row the selection is not maintained ( checkbox selection ) in the subcomponent.

Here is the example: https://codesandbox.io/s/condescending-rubin-ne6jw?file=/src/App.js

Can anyone help me to achieve ?