diff --git a/app/src/main/java/org/mozilla/fenix/components/toolbar/TabCounterToolbarButton.kt b/app/src/main/java/org/mozilla/fenix/components/toolbar/TabCounterToolbarButton.kt index b9c568089f..f0cb37eed1 100644 --- a/app/src/main/java/org/mozilla/fenix/components/toolbar/TabCounterToolbarButton.kt +++ b/app/src/main/java/org/mozilla/fenix/components/toolbar/TabCounterToolbarButton.kt @@ -28,14 +28,21 @@ class TabCounterToolbarButton( val view = TabCounter(parent.context).apply { reference = WeakReference(this) - setCount(sessionManager.sessions.count { - it.private == isPrivate - }) setOnClickListener { showTabs.invoke() } contentDescription = parent.context.getString(R.string.mozac_feature_tabs_toolbar_tabs_button) + + addOnAttachStateChangeListener(object : View.OnAttachStateChangeListener { + override fun onViewAttachedToWindow(v: View?) { + setCount(sessionManager.sessions.count { + it.private == isPrivate + }) + } + + override fun onViewDetachedFromWindow(v: View?) { /* no-op */ } + }) } // Set selectableItemBackgroundBorderless