1 files changed, 7 insertions, 7 deletions

diff --git a/recipes/element/webview.js b/recipes/element/webview.js
index 18bb7ad..c1fdea3 100644
--- a/recipes/element/webview.js
+++ b/recipes/element/webview.js
@@ -3,19 +3,19 @@ module.exports = Ferdi => {
     const badges = document.querySelectorAll('.mx_RoomSublist:not(.mx_RoomSublist_hidden) .mx_RoomSublist_badgeContainer');
 
     // Number of messages from People
-    let direct_count = 0;
-    if (badges.length > 0 && badges[0].children.length > 0) {
-      direct_count = parseInt(badges[0].textContent);
+    let directCount = 0;
+    if (badges.length > 0) {
+      directCount = Ferdi.safeParseInt(badges[0].textContent);
     }
 
     // Number of messages from Rooms
-    let indirect_count = 0;
-    if (badges.length > 1 && badges[1].children.length > 0) {
-      indirect_count = parseInt(badges[1].textContent);
+    let indirectCount = 0;
+    if (badges.length > 1) {
+      indirectCount = Ferdi.safeParseInt(badges[1].textContent);
     }
 
     // set Ferdi badge
-    Ferdi.setBadge(direct_count, indirect_count);
+    Ferdi.setBadge(directCount, indirectCount);
   }
 
   Ferdi.loop(getMessages);