Small fix to headers