Fixes refactoring mess. Still need to pass context from above to each light.