1 files changed, 11 insertions, 3 deletions

diff --git a/recipes/infomaniak-mail/webview.js b/recipes/infomaniak-mail/webview.js
index 04ac42d..f4e39dc 100644
--- a/recipes/infomaniak-mail/webview.js
+++ b/recipes/infomaniak-mail/webview.js
@@ -1,8 +1,16 @@
 module.exports = Ferdi => {
   const getMessages = () => {
-    const count = document.querySelector('.ws-tree-node-badge').innerText;
-    Ferdi.setBadge(count ? Number(count.substring(1, count.length - 1)) : 0);
-  }
+    const count = document.querySelector('.ws-tree-node-badge');
+    if (count) {
+      const countText = count.textContent;
+      if (countText) {
+        Ferdi.setBadge(
+          // eslint-disable-next-line unicorn/prefer-string-slice
+          count ? Number(countText.substring(1, countText.length - 1)) : 0,
+        );
+      }
+    }
+  };
 
   Ferdi.loop(getMessages);
 };