Fixed unit test following model change.