1 files changed, 5 insertions, 3 deletions
diff --git a/scrshot b/scrshot
index fe21abd..77174e6 100755
--- a/scrshot
+++ b/scrshot
@@ -7,7 +7,7 @@ output_text () {
 
 while getopts "hwxd:s:o:" o; do case "${o}" in
   w)
-    DMENU="bemenu"
+    DMENU="tofi"
     SCRSH="grim"
     ;;
   x)
@@ -21,10 +21,12 @@ esac done
 
 if [ -z "$DMENU" ]; then
   if [ -n "$WAYLAND_DISPLAY" ]; then
-    DMENU="bemenu"
+    DMENU="tofi"
+    DEFARGS="--prompt-text"
     SCRSH="grim"
   elif [ -n "$DISPLAY" ]; then
     DMENU="dmenu"
+    DEFARGS="-p"
     SCRSH="maim"
   else
     output_text "Stuck" "Can't tell if Wayland or X, what gives?"
@@ -41,7 +43,7 @@ sleep 0.1
 "$SCRSH" $SARG "$TEMPFILE"
 
 if [ -z "$FILENAME" ]; then
-  FILENAME="$(echo "$TEMPFILE" | $DMENU $DARGS -p "What to name the file?")"
+  FILENAME="$(echo "$TEMPFILE" | $DMENU $DARGS $DEFARGS "What to name the file?")"
   if [ -z "$FILENAME" ]; then
     rm -rf "$TEMPFILE"
     output_text "Deleted" "No filename selected, file deleted!"