diff --git a/plugins/contrib/treestyletabs.js b/plugins/contrib/treestyletabs.js
index ece387c4..ff623223 100644
--- a/plugins/contrib/treestyletabs.js
+++ b/plugins/contrib/treestyletabs.js
@@ -39,8 +39,8 @@ function fold_collapse_expand(collapse, children = false) {
     let tab = gBrowser.tabContainer.selectedItem;
 
     if (!TreeStyleTabService.hasChildTabs(tab)) {
-        let tab = TreeStyleTabService.getParentTab(tab);
-        collapse = TreeStyleTabService.isSubtreeCollapsed(tab);
+        tab = TreeStyleTabService.getParentTab(tab);
+        gBrowser.tabContainer.selectedIndex = gBrowser.tabContainer.getIndexOfItem(tab);
     }
 
     fold_collapse_expand_target(