add missing include to fix compile error