Merge pull request #408 from iamqizhao/master

fix a bug in test