修复bug

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