Change some code of test.