Little bit modify the test case