diff --git a/homeManagerModules/desktop/wayland/default.nix b/homeManagerModules/desktop/wayland/default.nix
index 640a9d59..48f5f796 100644
--- a/homeManagerModules/desktop/wayland/default.nix
+++ b/homeManagerModules/desktop/wayland/default.nix
@@ -38,6 +38,8 @@
       playerctld.enable = lib.mkDefault true;
       swayosd.enable = lib.mkDefault true;
     };
+    
+    systemd.user.services.swayosd.Install.WantedBy = [ "hyprland-session.target" "sway-session.target" ];
 
     xdg.portal = {
       enable = true;
diff --git a/homeManagerModules/services/swayidle/default.nix b/homeManagerModules/services/swayidle/default.nix
index d32535d8..8ae0e452 100644
--- a/homeManagerModules/services/swayidle/default.nix
+++ b/homeManagerModules/services/swayidle/default.nix
@@ -58,5 +58,7 @@ in {
           command = "${swaymsg} \"output * dpms off\"' resume '${swaymsg} \"output * dpms on\"";
         };
     };
+    
+    systemd.user.services.swayidle.Service.Restart = lib.mkForce  "on-failure";
   };
 }