Fixed an issue with the header file.