1 files changed, 12 insertions, 1 deletions
diff --git a/dmenu_books b/dmenu_books
index beb87cc..f24391b 100755
--- a/dmenu_books
+++ b/dmenu_books
@@ -5,12 +5,23 @@ printhelp () {
 }
 
 while getopts "hwxd:" o; do case "${o}" in
-  w) DMENU="tofi" ;;
+  w) DMENU="bemenu" ;;
   x) DMENU="dmenu" ;;
   d) ARGS="$ARGS$OPTARG " ;;
   *) printhelp ;;
 esac done
 
+if [ -z "$DMENU" ]; then
+  if [ -n "$WAYLAND_DISPLAY" ]; then
+    DMENU="bemenu"
+  elif [ -n "$DISPLAY" ]; then
+    DMENU="dmene"
+  else
+    print "Can't tell if Wayland or X; what gives?\n"
+    exit 1
+  fi
+fi
+
 BOOK=`ls ~/Documents/shared | $DMENU $ARGS`
 
 [ -z "$BOOK" ] && exit 1