1 files changed, 4 insertions, 5 deletions

diff --git a/uncompressed/slack/webview.js b/uncompressed/slack/webview.js
index 98c6136..174d81c 100644
--- a/uncompressed/slack/webview.js
+++ b/uncompressed/slack/webview.js
@@ -9,11 +9,11 @@ function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { de
 const getTeamIcon = function getTeamIcon(count = 0) {
   let countTeamIconCheck = count;
   let bgUrl = null;
-  const teamMenu = document.querySelector('#team_menu');
+  const teamMenu = document.querySelector('#team-menu-trigger');
 
   if (teamMenu) {
     teamMenu.click();
-    const icon = document.querySelector('.team_icon');
+    const icon = document.querySelector('.c-team_icon');
 
     if (icon) {
       bgUrl = window.getComputedStyle(icon, null).getPropertyValue('background-image');
@@ -22,8 +22,7 @@ const getTeamIcon = function getTeamIcon(count = 0) {
     }
 
     setTimeout(() => {
-      document.querySelector('.team_menu').remove();
-      document.querySelector('#msg_input .ql-editor').focus();
+      document.querySelector('.ReactModal__Overlay').click();
     }, 10);
   }
 
@@ -42,7 +41,7 @@ const SELECTOR_CHANNELS_UNREAD = '.p-channel_sidebar__channel--unread:not(.p-cha
 
 module.exports = Franz => {
   const getMessages = () => {
-    const directMessages = document.querySelectorAll(`${SELECTOR_CHANNELS_UNREAD} .p-channel_sidebar__badge`).length;
+    const directMessages = document.querySelectorAll(`${SELECTOR_CHANNELS_UNREAD} .p-channel_sidebar__badge, .p-channel_sidebar__link--unread`).length;
     const allMessages = document.querySelectorAll(SELECTOR_CHANNELS_UNREAD).length - directMessages;
     Franz.setBadge(directMessages, allMessages);
   };