1 files changed, 7 insertions, 3 deletions

diff --git a/recipes/mysms/webview.js b/recipes/mysms/webview.js
index 3f3415a..1878120 100644
--- a/recipes/mysms/webview.js
+++ b/recipes/mysms/webview.js
@@ -1,10 +1,14 @@
 module.exports = Ferdi => {
   const getMessages = () => {
-    const elements = document.getElementsByClassName('unread');
+    const elements = document.querySelectorAll('.unread');
 
     let count = 0;
-    for (let i = 0; i < elements.length; i++) {
-      if (Ferdi.safeParseInt(elements[i].innerText.replace(/[^0-9.]/g, '')) > 0) {
+    for (const element of elements) {
+      if (
+        Ferdi.safeParseInt(
+          element.textContent && element.textContent.replace(/[^\d.]/g, ''),
+        ) > 0
+      ) {
         count++; // count 1 per channel with messages
       }
     }