From beb7d09aeeadb0f78502fba66792f065abada2e8 Mon Sep 17 00:00:00 2001
From: Adorian Doran <adorian@esevo.ro>
Date: Sat, 22 Nov 2025 13:35:36 +0200
Subject: [PATCH] client/tree context menu: improve handling the case when the
 note color picker is not available

---
 apps/client/src/menus/tree_context_menu.ts | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

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" },