modify the test function