TreeView - deselect child on collpase

Hi,

Has any one figured out a way to de-select child nodes (or select the collapsed parent) in the Blazor treeView when a node is collapsed? There's an OnExpand event, but no OnCollapse, so I'm struggling to figure a way to do this.

Thanks