diff --git a/webview.js b/webview.js
index 34ce730..383446f 100644
--- a/webview.js
+++ b/webview.js
@@ -3,8 +3,10 @@ import path from 'path';
 module.exports = (Franz) => {
   const getMessages = function getMessages() {
     // get unread messages
-    const direct = document.querySelectorAll('[class^="guildsWrapper"] [class*="badge"]').length;
-    const indirect = document.querySelectorAll('[class^="guildsWrapper"] [class^="guild-"]+[class*="unread-"]').length;
+    const serverbar = document.querySelector('[class^=unreadMentionsIndicatorTop] + div');
+    if (!serverbar) return;
+    const direct = serverbar.querySelectorAll('[class*="lowerBadge"]').length;
+    const indirect = 0;
 
     // set Franz badge
     Franz.setBadge(direct, indirect);