diff --git a/modules/services/wayneko/module.nix b/modules/services/wayneko/module.nix
index 722fa24..d3385e5 100644
--- a/modules/services/wayneko/module.nix
+++ b/modules/services/wayneko/module.nix
@@ -9,8 +9,11 @@ in {
   systemd.user.services.wayneko = {
     description = "Wayneko, as a systemd service";
 
+    # makes the graphical session start this service when it starts
     wantedBy = ["graphical-session.target"];
-    wants = ["graphical-session.target"];
+    # when graphical session restarts or gets stopped, this also gets restarted/stopped.
+    partOf = ["graphical-session.target"];
+    # gets started only after graphical session
     after = ["graphical-session.target"];
 
     serviceConfig = {