diff --git a/modules/lock-action.nix b/modules/lock-action.nix
index 69cc8bc..1bf124c 100644
--- a/modules/lock-action.nix
+++ b/modules/lock-action.nix
@@ -41,8 +41,7 @@ in
         SSH_AUTH_SOCK = "fake";
       };
 
-      wants = [ "graphical-session.target" ];
-      partOf = [ "graphical-session.target" ];
+      wantedBy = [ "graphical-session.target" ];
       after = [ "graphical-session.target" ];
     };
   };