diff --git a/templates/.config/swayidle/config b/templates/.config/swayidle/config
index 4722b5d..e66b25a 100644
--- a/templates/.config/swayidle/config
+++ b/templates/.config/swayidle/config
@@ -1 +1,2 @@
-timeout 10 'pgrep swaylock && swaymsg "output * dpms off"' resume 'pgrep swaylock && swaymsg "output * dpms on"'
+timeout 10 'pgrep swaylock && ${host.get('display-off-cmd', 'swaymsg "output * dpms off"')}' \
+resume 'pgrep swaylock && ${host.get('display-on-cmd', 'swaymsg "output * dpms on"')}'