代码小优化

Signed-off-by: TZ <atian25@qq.com>