diff --git a/modules/security.nix b/modules/security.nix
index cefac61..165ae3f 100644
--- a/modules/security.nix
+++ b/modules/security.nix
@@ -35,8 +35,7 @@
       lib.pipe ''
         ${pkgs.sway}/bin/swaymsg output "*" dpms off
         ${pkgs.systemd}/bin/systemd-inhibit --why="Already locked" --what=idle --who="lock script" ${config.security.wrapperDir}/physlock -s -p "${lockMessage}"
-        while [ $(${pkgs.sway}/bin/swaymsg -t get_seats | ${pkgs.jq}/bin/jq "[.[] | .capabilities] | max") -eq 0 ]; do ${pkgs.coreutils}/bin/sleep 0.1; done
-        ${pkgs.sway}/bin/swaymsg output "*" dpms on
+        while [ $(${pkgs.sway}/bin/swaymsg -t get_outputs | ${pkgs.jq}/bin/jq "[.[] | .dpms] | any") = "false" ]; do ${pkgs.coreutils}/bin/sleep 0.1; ${pkgs.sway}/bin/swaymsg output "*" dpms on; done
       '' [
         (lib.splitString "\n")
         (lib.filter (line: line != ""))