Remove the test function