diff --git a/neofetch b/neofetch
index a0ac4f44..616b13e9 100755
--- a/neofetch
+++ b/neofetch
@@ -2264,8 +2264,8 @@ getimage () {
         "wall") getwallpaper ;;
         "ascii") getascii; return ;;
         *)
-            if [ "${image: -1}" == "/" ]; then
-                files=("$image"*.png "$image"*.jpg)
+            if [ -d "$image" ]; then
+                files=("$image"*.{png,jpg,jpeg})
                 img="$(printf "%s" "${files[RANDOM % (${#files[@]} - 1)]}")"
             else
                 img="$image"