fix bug
Signed-off-by: IcePlus <zhengjiabin.tj@gmail.com>