diff --git a/modules/niri/default.nix b/modules/niri/default.nix
index c222dcc..92650ee 100644
--- a/modules/niri/default.nix
+++ b/modules/niri/default.nix
@@ -46,7 +46,10 @@
       services.swayidle = {
         enable = true;
         events = [
-          { event = "before-sleep"; command = "${pkgs.swaylock}/bin/swaylock"; }
+          { event = "before-sleep"; command = "swaylock"; }
+        ];
+        timeouts = [
+          { timeout = 5; command = "swaylock"; }
         ];
       };