Test updated for new version of LightModels