fix bug.

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