Fix header bugs