diff --git a/bg-feh b/bg-feh
index 23d0975..4e56fa5 100755
--- a/bg-feh
+++ b/bg-feh
@@ -1,13 +1,16 @@
 #!/bin/sh
 
-[ -f "$1" ] || exit 1
+[ -e "$1" ] || exit 1
 
 if [ -d $1 ]; then
   bg=$(ls $1 | shuf -n 1)
+  bg="$1/$bg"
 elif [ -f $1 ]; then
   bg=$1
 else
   exit 1
 fi
 
+echo $bg
+
 feh --bg-fill "$bg"