fix model test to match model change