Fix header bug