diff --git a/platform/collaboration-tools/src/com/intellij/collaboration/ui/codereview/editor/action/CodeReviewInEditorToolbarActionGroup.kt b/platform/collaboration-tools/src/com/intellij/collaboration/ui/codereview/editor/action/CodeReviewInEditorToolbarActionGroup.kt
index 31a7ebf37d68..7250756daa87 100644
--- a/platform/collaboration-tools/src/com/intellij/collaboration/ui/codereview/editor/action/CodeReviewInEditorToolbarActionGroup.kt
+++ b/platform/collaboration-tools/src/com/intellij/collaboration/ui/codereview/editor/action/CodeReviewInEditorToolbarActionGroup.kt
@@ -36,14 +36,11 @@ class CodeReviewInEditorToolbarActionGroup(private val vm: CodeReviewInEditorVie
   override fun getChildren(e: AnActionEvent?): Array<AnAction> =
     arrayOf(updateAction, Separator.create(), disableReviewAction, hideResolvedAction, showAllAction)
 
-  override fun displayTextInToolbar(): Boolean = true
-
-  override fun useSmallerFontForTextInToolbar(): Boolean = true
-
   init {
     with(templatePresentation) {
       isPopupGroup = true
       putClientProperty(ActionUtil.HIDE_DROPDOWN_ICON, true)
+      putClientProperty(ActionUtil.USE_SMALL_FONT_IN_TOOLBAR, true)
       description = CollaborationToolsBundle.message("review.editor.mode.description.title")
       val tooltip = HelpTooltip()
         .setTitle(CollaborationToolsBundle.message("review.editor.mode.description.title"))
@@ -58,10 +55,12 @@ class CodeReviewInEditorToolbarActionGroup(private val vm: CodeReviewInEditorVie
     with(e.presentation) {
       if (shown) {
         text = CollaborationToolsBundle.message("review.editor.mode.title")
+        putClientProperty(ActionUtil.SHOW_TEXT_IN_TOOLBAR, true)
         icon = if (synced) null else getWarningIcon()
       }
       else {
         text = null
+        putClientProperty(ActionUtil.SHOW_TEXT_IN_TOOLBAR, false)
         icon = AllIcons.Actions.Preview
       }
     }