Lightbox bug fix