Fix specs after model changes