diff --git a/konawall/gui.py b/konawall/gui.py
index 8d27530..7cc3d5e 100755
--- a/konawall/gui.py
+++ b/konawall/gui.py
@@ -71,7 +71,7 @@ class Konawall(wx.adv.TaskBarIcon):
         # Set up the taskbar icon, menu, bindings, ...
         icon = self.generate_icon()
         self.SetIcon(icon, self.title_string)
-        if self.environment in ["hyprland", "gnome"]:
+        if self.environment in ["hyprland", "niri", "gnome"]:
             import pystray
             def setup(self):
                 self.visible = True