Minor fix in header file