1 files changed, 8 insertions, 10 deletions

diff --git a/recipes/proton-mail/webview.js b/recipes/proton-mail/webview.js
index 395b779..89e5dab 100644
--- a/recipes/proton-mail/webview.js
+++ b/recipes/proton-mail/webview.js
@@ -1,15 +1,13 @@
 module.exports = Ferdi => {
   const getMessages = () => {
-    const element = document.querySelector('.navigationItem-counter');
-    if (!element) {
-      return;
-    }
-    const text = element.textContent;
-    if (text) {
-      // eslint-disable-next-line unicorn/prefer-string-slice
-      const count = Ferdi.safeParseInt(text.substring(1, text.length - 1));
-      Ferdi.setBadge(count);
-    }
+    let unreadCount = 0;
+    // Loop over all displayed counters and take the highest one (from the "All Mail" folder)
+    document.querySelectorAll('.navigation-counter-item').forEach(counterElement => {
+      const unreadCounter = Ferdi.safeParseInt(counterElement.textContent);
+      unreadCount = Math.max(unreadCount, unreadCounter);
+    });
+
+    Ferdi.setBadge(unreadCount);
   };
 
   Ferdi.loop(getMessages);