Added missing include & fix for compiler warning