Fixed the header file