修复bug

Signed-off-by: wanghao <wanghao@wanghao-PC>