Fixed an issue with the header files