Fix compile erorr caused by upstream API change