修复代码

Signed-off-by: wushun <welshu@qq.com>