Image highlight not on right image

After opening and closing an image from a comment while viewing an image in the gallery (the type that pops up in the shadow-box), the yellow highlighting box on the image selector (top-right 3x2 thumbnail block) does not update to the current image. This is most noticeable when on the last image in a set of six because when I go to the next image, the image selector is still on the old set of six and not on the new set of six.