diff --git a/main.go b/main.go
index 855b2e5..42a72b3 100644
--- a/main.go
+++ b/main.go
@@ -92,9 +92,14 @@ func newBrowse(file string) {
 		return
 	}
 	directoryState.images = []string{}
+	i := 0
 	for _, v := range dirfiles {
 		if filetypes.IsImage(v) {
 			directoryState.images = append(directoryState.images, v.Name())
+			if filepath.Join(directoryState.dir, v.Name()) == file {
+				directoryState.i = i
+			}
+			i++
 		}
 	}
 	updateImage(file)