Update test due to model changes.