diff --git a/scripts/i3wm-config-gen.sh b/scripts/i3wm-config-gen.sh
index 28e24ee..d38ff09 100755
--- a/scripts/i3wm-config-gen.sh
+++ b/scripts/i3wm-config-gen.sh
@@ -5,4 +5,8 @@
 
 cat ~/.config/i3/shared.conf ~/.config/i3/${computer}.conf > ~/.config/i3/config
 
-i3-msg reload
+if command -v i3-msg &> /dev/null; then
+    i3-msg reload
+elif command -v swaymsg &> /dev/null; then
+    swaymsg reload
+fi