JS fix for lightbox