fix tests
Merge pull request #63 from mohayonao/fix-tests

fix tests