fix unittest failure
Merge pull request #26 from yeshunping/test_fix

fix unittest failure