Add missing include in wrapper class