fixed bug in header file