diff --git a/modules/users.nix b/modules/users.nix
index 386a3ef..9503a53 100644
--- a/modules/users.nix
+++ b/modules/users.nix
@@ -37,7 +37,16 @@
     "/run/user/1000";
 
   services.udev.extraRules = ''
-    ACTION=="remove", ATTRS{idVendor}=="1050", RUN+="${pkgs.systemd}/bin/systemctl suspend"'';
+    ACTION=="remove", ATTRS{idVendor}=="1050", RUN+="${pkgs.systemd}/bin/systemctl start swaylock"'';
+
+  systemd.services.swaylock = {
+    description = "Lock the screen";
+    serviceConfig.User = "balsoft";
+    environment.XDG_RUNTIME_DIR = "/run/user/1000";
+    serviceConfig.Type = "oneshot";
+    path = [pkgs.swaylock];
+    script = "swaylock -c ${builtins.substring 1 7 config.themes.colors.bg}";
+  };
 
   security.pam.u2f = {
     control = "sufficient";