1 files changed, 3 insertions, 6 deletions

diff --git a/uncompressed/zalo/webview.js b/uncompressed/zalo/webview.js
index e20c6b5..0f360c3 100644
--- a/uncompressed/zalo/webview.js
+++ b/uncompressed/zalo/webview.js
@@ -3,12 +3,9 @@
 module.exports = Franz => {
   const getMessages = function getMessages() {
     var unread = 0
-    const notificationBadge = document.getElementsByClassName('notification-badge')[0]
-    if (notificationBadge != undefined) {
-        unread = notificationBadge.innerText;
-    }
-    Franz.setBadge(parseInt(unread, 10));
+    const notificationBadge = document.getElementsByClassName('tab-red-dot').length;
+    Franz.setBadge(notificationBadge);
   };
 
   Franz.loop(getMessages);
-};
\ No newline at end of file
+};