small improvement

Signed-off-by: Zhao, Zhongjie <123378840@qq.com>