Minor fix for header