2 files changed, 10 insertions, 6 deletions

diff --git a/recipes/telegram/package.json b/recipes/telegram/package.json
index e0a7900..ee90470 100644
--- a/recipes/telegram/package.json
+++ b/recipes/telegram/package.json
@@ -1,7 +1,7 @@
 {
   "id": "telegram",
   "name": "Telegram",
-  "version": "3.1.0",
+  "version": "3.1.1",
   "license": "MIT",
   "config": {
     "serviceURL": "https://web.telegram.org",
diff --git a/recipes/telegram/webview.js b/recipes/telegram/webview.js
index 8da63d1..db8c8d7 100644
--- a/recipes/telegram/webview.js
+++ b/recipes/telegram/webview.js
@@ -8,11 +8,15 @@ module.exports = Franz => {
     let count_sec = 0;
     const elements = document.querySelectorAll('.rp');
     for (let i = 0; i < elements.length; i += 1) {
-      if (elements[i].querySelector('.dialog-subtitle-badge') && (!isNaN(parseInt(elements[i].querySelector('.dialog-subtitle-badge').innerText)))) {
-        if (parseInt(elements[i].querySelector('.dialog-subtitle-badge').innerText) != '' && (elements[i].dataset.peerId > 0)) {
-          count = +count + parseInt(elements[i].querySelector('.dialog-subtitle-badge').innerText);
-        } else {
-          count_sec = +count_sec + parseInt(elements[i].querySelector('.dialog-subtitle-badge').innerText);
+      const subtitleBadge = elements[i].querySelector('.dialog-subtitle-badge');
+      if (subtitleBadge) {
+        const parsedValue = parseInt(subtitleBadge.innerText);
+        if (!isNaN(parsedValue)) {
+          if (elements[i].dataset.peerId > 0) {
+            count += parsedValue;
+          } else {
+            count_sec += parsedValue;
+          }
         }
       }
     }