Fixed a missing declaration in the .h files