diff --git a/apps/client/src/menus/tree_context_menu.ts b/apps/client/src/menus/tree_context_menu.ts
index 19e6f4e16..901a16559 100644
--- a/apps/client/src/menus/tree_context_menu.ts
+++ b/apps/client/src/menus/tree_context_menu.ts
@@ -244,16 +244,12 @@ export default class TreeContextMenu implements SelectMenuItemEventListener<Tree
 
             { kind: "separator"},
 
-            {
+            (notOptionsOrHelp && selectedNotes.length === 1) ? {
                 kind: "custom",
                 componentFn: () => {
-                    if (notOptionsOrHelp && selectedNotes.length === 1) {
-                        return NoteColorPicker({note});
-                    } else {
-                        return null;
-                    }
+                    return NoteColorPicker({note});
                 }
-            },
+            } : null,
 
             { kind: "separator" },