bug fix

Signed-off-by: LiYuancen <lyc1116@foxmail.com>