Added missing header file reference