diff --git a/src/bitmessageqt/__init__.py b/src/bitmessageqt/__init__.py
index a2968fe1..57a59cbc 100644
--- a/src/bitmessageqt/__init__.py
+++ b/src/bitmessageqt/__init__.py
@@ -511,10 +511,10 @@ class MyForm(settingsmixin.SMainWindow):
             isMaillinglist = shared.safeConfigGetBoolean(
                 toAddress, 'mailinglist')
 
-            if tab == 'messages':
+            if treeWidget == self.ui.treeWidgetYourIdentities:
                 if isChan:
                     continue
-            elif tab == 'chan':
+            elif treeWidget == self.ui.treeWidgetChans:
                 if not isChan:
                     continue
 
@@ -532,7 +532,7 @@ class MyForm(settingsmixin.SMainWindow):
             total += cnt
             if toaddress in db and folder in db[toaddress]:
                 db[toaddress][folder] = cnt
-        if tab == "messages":
+        if treeWidget == self.ui.treeWidgetYourIdentities:
             db[None] = {}
             db[None]["inbox"] = total
             db[None]["new"] = total