diff --git a/other/scripts/desktop/shell-startup.nu b/other/scripts/desktop/shell-startup.nu
index d4b9a51..245927b 100644
--- a/other/scripts/desktop/shell-startup.nu
+++ b/other/scripts/desktop/shell-startup.nu
@@ -17,14 +17,14 @@ def flag [
   mut cols = $colors | wrap color | insert width $col_size | flatten;
   let last = ($colors | length) - 1;
 
-  $cols = if not ($rest | is even) {
+  if not ($rest | is even) {
     $rest = $rest - 1;
 
-    (if not ($colors | length | is even) {
+    $cols = (if not ($colors | length | is even) {
       $cols | update (($colors | length) / 2 | math floor) { $in | update width {|w| ($w.width + 1)}}
     } else {
       $cols | update $last { $in | update width {|w| ($w.width + 1)}}
-    })
+    });
   };
 
   std assert ($rest | is even);