Add missing #include to fix compiler warning