1 files changed, 3 insertions, 3 deletions
diff --git a/yts b/yts
index 7401761..079c628 100755
--- a/yts
+++ b/yts
@@ -2,7 +2,7 @@
 
 url_yts="https://yts.mx"
 proxy_yts="https://yts.torrentbay.to"
-menu="dmenu"
+menu="wmenu"
 
 die()
 {
@@ -69,8 +69,8 @@ main()
 	out=
 	query=
 
-	[ -z "$DISPLAY" ] &&
-		menu="wmenu"
+	[ -z "$WAYLAND_DISPLAY" ] &&
+		menu="dmenu"
 	if [ "$#" -gt 0 ] && [ "$1" = "-o" ]
 	then
 		out=true