Fixed header file issue