Change model and correct test.