Collapse, Expand all button and tree state is disconnected.

[DimalChandrasiri]

Describe the Issue:
When the permission tree is collapsed using the arrow the button still shows collapse all even though the tree is already collapsed. Refer image.