fix header for test code