modify test function