diff --git a/.config/yadm/bootstrap.d/_install_desktop_env.sh b/.config/yadm/bootstrap.d/_install_desktop_env.sh
index 7d2d8a6..7efacec 100755
--- a/.config/yadm/bootstrap.d/_install_desktop_env.sh
+++ b/.config/yadm/bootstrap.d/_install_desktop_env.sh
@@ -1,5 +1,9 @@
 #!/bin/env bash
 
+if [ -z ${XDG_SESSION_DESKTOP+x} ]; then
+  read -p "Which desktop environment do you want to install? " -r XDG_SESSION_DESKTOP
+fi
+
 echo "Installing dependencies for $XDG_SESSION_DESKTOP"
 
 if [ $XDG_SESSION_DESKTOP = "i3" ]; then
@@ -8,11 +12,11 @@ if [ $XDG_SESSION_DESKTOP = "i3" ]; then
     dunst \
     pulseaudio-control \
     nitrogen
-
 fi
 
 if [ $XDG_SESSION_DESKTOP = "Hyprland" ]; then
-  yay -S --needed waybar \
+  yay -S --needed hyprland \
+    waybar \
     hyprlock \
     swww \
     hyprswitch \
@@ -25,5 +29,4 @@ if [ $XDG_SESSION_DESKTOP = "Hyprland" ]; then
     wl-clipboard \
     grim \
     slurp
-
 fi