1 files changed, 9 insertions, 1 deletions

diff --git a/recipes/msteams/webview.js b/recipes/msteams/webview.js
index 92af3a4..b343196 100644
--- a/recipes/msteams/webview.js
+++ b/recipes/msteams/webview.js
@@ -7,9 +7,17 @@ const _path = _interopRequireDefault(require('path'));
 module.exports = Ferdium => {
   const getMessages = () => {
     let messages = 0;
-    const badges = document.querySelectorAll(
+
+    const isTeamsV2 = window.location.href.includes('/v2/');
+
+    let badges = document.querySelectorAll(
       '.activity-badge.dot-activity-badge .activity-badge',
     );
+
+    if (isTeamsV2) {
+      badges = document.querySelectorAll('.fui-Badge');
+    }
+
     if (badges) {
       Array.prototype.forEach.call(badges, badge => {
         messages += Ferdium.safeParseInt(badge.textContent);