Small fix for lights