recalculate sort to handle favourites changes

1 files changed, 7 insertions, 7 deletions

diff --git a/main.lua b/main.lua
index 2273720..b91a458 100644
--- a/main.lua
+++ b/main.lua
@@ -948,16 +948,16 @@ local function toggle_menu_sort()
 
 	local key = menu.type == 'search' and 'search_options' or 'options'
 
-	if not menu['sorted_' .. key] then
+	menu.sorted = not menu.sorted
+	if menu.sorted then
 		menu['orig_' .. key] = menu[key]
-		menu['sorted_' .. key] = util.copy_table(menu[key])
-		sort_options(menu['sorted_' .. key])
+		menu[key] = util.copy_table(menu[key])
+		sort_options(menu[key])
+	else
+		menu[key] = menu['orig_' .. key]
+		menu['orig_' .. key] = nil
 	end
 
-	menu.sorted = not menu.sorted
-	menu[key] = menu.sorted and menu['sorted_' .. key] or
-		menu['orig_' .. key]
-
 	if menu.type == 'search' then
 		update_search_matches()
 	end