修复bug

Signed-off-by: taurus <xxh0755@gmail.com>