@yifan-gu Why you need to wrap this?