CheckBox header not marked as checked with collapse tree grid

please look into following images let me know if you have any solution for it,

when I click on select all header checkbox (last column of grid) at that time if grid having row in collapse mode at that time all the record get selected but header checkbox remains unchecked but when I expand that particular record grid then header checkbox get selected please help me with it.

how can we get checked header checkbox while grid data in collapse mode.

enter image description here

enter image description here