fix the test file and model