Fix the protocol lib integration. Thanks @aadnk