fix the model test