fix lightbox functionality