简化代码

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