Fix infinite lights bug