[Zhenghan] bug fixed