Fix collapse all rightclick option being annoying in its behavior. Now it's *smart*!

1 files changed, 4 insertions, 0 deletions

diff --git a/src/courtroom.cpp b/src/courtroom.cpp
index 26648372..d5dd6565 100644
--- a/src/courtroom.cpp
+++ b/src/courtroom.cpp
@@ -3917,6 +3917,10 @@ void Courtroom::music_list_expand_all()
 void Courtroom::music_list_collapse_all()
 {
   ui_music_list->collapseAll();
+  QTreeWidgetItem *current = ui_music_list->selectedItems()[0];
+  if (current->parent() != nullptr)
+    current = current->parent();
+  ui_music_list->setCurrentItem(current);
 }
 
 void Courtroom::on_area_list_double_clicked(QTreeWidgetItem *p_item, int column)