Fix header

I forgot to commit the header changes