Remove the test code from the header