Add missed header file