diff --git a/plugins/contrib/treestyletabs.js b/plugins/contrib/treestyletabs.js
index 216631d4..2e8fd870 100644
--- a/plugins/contrib/treestyletabs.js
+++ b/plugins/contrib/treestyletabs.js
@@ -50,8 +50,8 @@ function fold_collapse_expand_toggle(children = false) {
         );
 }
 
-function create_command_and_mapping(command, description, funcref, mapping) {
-    group.commands.add([command], description, funcref, {}, true );
+function create_command_and_mapping(command, description, funcref, mapping, command_option = {}) {
+    group.commands.add([command], description, funcref, command_option, true );
     if (mapping != "")
         group.mappings.add([modes.NORMAL], [mapping], description, funcref);
 }