fix badge for Instagram (#468)

2 files changed, 5 insertions, 3 deletions

diff --git a/recipes/instagram-direct-messages/package.json b/recipes/instagram-direct-messages/package.json
index 41aa129..ab94fae 100644
--- a/recipes/instagram-direct-messages/package.json
+++ b/recipes/instagram-direct-messages/package.json
@@ -1,7 +1,7 @@
 {
   "id": "instagram-direct-messages",
   "name": "Instagram Direct Messages",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "license": "MIT",
   "config": {
     "serviceURL": "https://www.instagram.com/direct/inbox/",
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,
     );
   };