fix model for test data