Fixed lightbox issue by changing selector