fix bug for test code.
Signed-off-by: gqjjqg <qijiang.guo@gmail.com>