Fix #2 - Lightbox working only in edit mode