fix badge for Instagram (#468)

1 files changed, 4 insertions, 2 deletions

diff --git a/recipes/instagram-direct-messages/webview.js b/recipes/instagram-direct-messages/webview.js
index 01187b8..eb33ed1 100644
--- a/recipes/instagram-direct-messages/webview.js
+++ b/recipes/instagram-direct-messages/webview.js
@@ -12,9 +12,11 @@ setInterval(() => {
 
 module.exports = Ferdium => {
   const getMessages = () => {
-    const element = document.querySelector('a[href^="/direct/inbox"]');
+    const element = document.querySelector('a[href^="/direct/inbox"] span');
     Ferdium.setBadge(
-      element.textContent ? Ferdium.safeParseInt(element.textContent) : 0,
+      element && element.textContent
+        ? Ferdium.safeParseInt(element.textContent)
+        : 0,
     );
   };