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

fix tests