修复bug

Signed-off-by: chenchong8806@gmail.com <chenchong8806@gmail.com>