fix bug
Signed-off-by: qijiang.guo <qijiang.guo@gmail.com>