diff --git a/setup b/setup
index 799b6b9..f52f5dd 100755
--- a/setup
+++ b/setup
@@ -688,7 +688,8 @@ sed -i 's/workspaces button.focused/workspaces button.active/' ~/.config/waybar/
 sed -i 's/font-family: FontAwesome, Roboto, Helvetica, Arial, sans-serif;/font-family: Terminess Nerd Font, FontAwesome, Roboto, Helvetica, Arial, sans-serif;/' ~/.config/waybar/style.css
 sed -i 's/sway/hyprland/' ~/.config/waybar/config.jsonc
 sed -i 's/"idle_inhibitor",//' ~/.config/waybar/config.jsonc
-sed -i 's/"custom/power",//' ~/.config/waybar/config.jsonc
+sed -i 's/"tray",/"tray"/' ~/.config/waybar/config.jsonc
+sed -i 's|"custom/power"||' ~/.config/waybar/config.jsonc
 
 
 #######################