Fix issue with public header file