diff --git a/other/scripts/desktop/shell-startup.nu b/other/scripts/desktop/shell-startup.nu
index af5e93a..d4b9a51 100644
--- a/other/scripts/desktop/shell-startup.nu
+++ b/other/scripts/desktop/shell-startup.nu
@@ -2,8 +2,6 @@ def "is even" [] {
   $in mod 2 == 0
 }
 
-# TODO: flag stuff as module to be used in other stuff in shell
-
 const TRANS = [ 5BCEFA F5A9B8 FFFFFF F5A9B8 5BCEFA ];
 const LESBIAN = [ D52D00 EF7627 FF9A56 FFFFFF D162A4 B55690 A30262 ];
 const ENBY = [ FCF434 FFFFFF 9C59D1 2C2C2C ];
@@ -19,14 +17,14 @@ def flag [
   mut cols = $colors | wrap color | insert width $col_size | flatten;
   let last = ($colors | length) - 1;
 
-  if not ($rest | is even) {
+  $cols = if not ($rest | is even) {
     $rest = $rest - 1;
 
-    $cols = (if not ($colors | length | is even) {
+    (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);