Collapse sections when Custom Sorting

I think sections appeared when sorting by custom criteria should be collapsible.

There is an arrow already, so please make it work.