1 files changed, 4 insertions, 2 deletions

diff --git a/recipes/nextcloud/webview.js b/recipes/nextcloud/webview.js
index 786f7a8..cb13e02 100644
--- a/recipes/nextcloud/webview.js
+++ b/recipes/nextcloud/webview.js
@@ -1,7 +1,9 @@
 module.exports = Ferdium => {
   const getMessages = () => {
-    const direct = document.querySelectorAll(
-      '.notifications .notification-container .notification-wrapper li .notification',
+    const direct = document.querySelector(
+      '#header-menu-notifications, .notifications'
+    ).querySelectorAll(
+      '.notification-container .notification-wrapper .notification'
     ).length;
 
     Ferdium.setBadge(direct);