Modify the code of sample test.