From 40f1948189c2183517a72a98d8ded6748b71870b Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Sun, 30 Apr 2023 20:30:07 +0530
Subject: [PATCH] Cleanup previous PR

The sort button used by the grid view still needs update_menu to take a
menu parameter
---
 src/calibre/gui2/actions/sort.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/calibre/gui2/actions/sort.py b/src/calibre/gui2/actions/sort.py
index fe104ee2d7..71b285d24a 100644
--- a/src/calibre/gui2/actions/sort.py
+++ b/src/calibre/gui2/actions/sort.py
@@ -98,8 +98,8 @@ class SortByAction(InterfaceAction):
             w.setStyleSheet('QToolButton::menu-button {image: none; }'
                             'QToolButton::menu-arrow {image: none; }')
 
-    def update_menu(self):
-        menu = self.qaction.menu()
+    def update_menu(self, menu=None):
+        menu = menu or self.qaction.menu()
         for action in menu.actions():
             if hasattr(action, 'sort_requested'):
                 action.sort_requested.disconnect()