* fix

* test

* remove setup change

* fix

* fix
Co-authored-by: Ntangwei12 <tangwei12@baidu.com>