diff --git a/home/wm/hyprland/default.nix b/home/wm/hyprland/default.nix
index b1cc884..bfe5e36 100644
--- a/home/wm/hyprland/default.nix
+++ b/home/wm/hyprland/default.nix
@@ -5,7 +5,7 @@ let
   hyprctl = "${config.wayland.windowManager.hyprland.package}/bin/hyprctl";
   kitty = "${pkgs.kitty}/bin/kitty";
   playerctl = "${pkgs.playerctl}/bin/playerctl";
-  rofi = "${pkgs.rofi-wayland}/bin/rofi";
+  rofi = "${config.programs.rofi.package}/bin/rofi";
   slurp = "${pkgs.slurp}/bin/slurp";
   swaylock = "${pkgs.swaylock}/bin/swaylock";
   tmux = "${pkgs.tmux}/bin/tmux";
diff --git a/home/wm/sway/default.nix b/home/wm/sway/default.nix
index d1d4138..d510f8e 100644
--- a/home/wm/sway/default.nix
+++ b/home/wm/sway/default.nix
@@ -77,7 +77,7 @@
     lib.mkOptionDefault {
       "${modifier}+Return" = "exec ${pkgs.kitty}/bin/kitty ${pkgs.tmux}/bin/tmux new";
       "${modifier}+Shift+Return" = "exec ${pkgs.kitty}/bin/kitty";
-      "${modifier}+d" = "exec ${pkgs.rofi-wayland}/bin/rofi -show drun";
+      "${modifier}+d" = "exec ${config.programs.rofi.package}/bin/rofi -show drun";
       "${modifier}+Shift+c" = "reload";
       "${modifier}+l" = "exec ${pkgs.swaylock}/bin/swaylock -f -i /tmp/lockpaper.jpg";
       "${modifier}+Shift+w" = "exec ~/.config/scripts/chpaper.sh";