fix bug


Signed-off-by: hanzheng <hzroom110@163.com>