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