Fixed compile error from missing include