Fix bug in header