confusing section folding toggle behavior

[bart9h]

In the status screen (the one you arrive with :G) you can open and close the diff for each file with =. You can also open and close the diffs for all files of a section at once by hitting = in the section header, right?

Not quite. Pressing = in the section header actually toggles the diff display for each file. If I have files A, B, C, D and E; individually open the diff for B, C and E; and then decide to close all diffs, by pressing = on the section header, it actually closes the diff for B, C and E but opens the diff for A and D.

The behavior I would expect is: if there is any diff opened, then close all; else, open all.

PS: thanks a lot for this wonderful plugin, it's a must-have to me