Added argument to match pandora_model changes