fix bug


Signed-off-by: leiddx <leiddx@vip.qq.com>