fix one test code