Fixed the compiler error (missing include)