Fixed a compilation error(missing include)