1 files changed, 3 insertions, 5 deletions

diff --git a/recipes/telegram-react/webview.js b/recipes/telegram-react/webview.js
index 223e914..8bc3839 100644
--- a/recipes/telegram-react/webview.js
+++ b/recipes/telegram-react/webview.js
@@ -1,15 +1,13 @@
 "use strict";
 
-const path = require("path");
-
 module.exports = Franz => {
   const getMessages = function getMessages() {
     let count = 0;
-    const elements = document.querySelectorAll(".dialog-badge:not(.dialog-badge-muted)");
+    const elements = document.querySelectorAll(".chatlist > li:not(.is-muted)");
     if (elements) {
       for (let i = 0; i < elements.length; i += 1) {
-        if (elements[i].querySelector("span").innerHTML !== 0) {
-          count += parseInt(elements[i].querySelector("span").innerHTML);
+        if (elements[i].querySelector(".unread") && elements[i].querySelector(".unread").innerHTML !== 0) {
+          count += parseInt(elements[i].querySelector(".unread").innerHTML);
         }
       }
     }