Fix the header to match the source