Small header fix