modify the test a little bit

Signed-off-by: Xiaofeng QU <quxiaofeng.at.polyu@gmail.com>