diff --git a/modules/home-manager/default.nix b/modules/home-manager/default.nix
index 76a4f48..0dd2db1 100644
--- a/modules/home-manager/default.nix
+++ b/modules/home-manager/default.nix
@@ -64,6 +64,8 @@
     pkgs.btop
     pkgs.zoom-us
     pkgs.pavucontrol
+    pkgs.wlogout
+    pkgs.swaylock
     inputs.kb-gui.packages.${pkgs.system}.kb
 
     # # You can also create simple shell scripts directly inside your
diff --git a/modules/rice/waybar/default.nix b/modules/rice/waybar/default.nix
index 05ec5c3..a4a3ebb 100644
--- a/modules/rice/waybar/default.nix
+++ b/modules/rice/waybar/default.nix
@@ -66,7 +66,7 @@ in {
         "custom/search" = {
           format = " ";
           tooltip = false;
-          on-click = "${pkgs.tofi}/bin/tofi-drun";
+          on-click = "${pkgs.wofi}/bin/wofi --show drun";
         };
 
         "custom/weather" = {