fix failing test because of model change