Modify test method