Modify test_helper