Fix light calculations