Header bug fix