fix tests
Merge pull request #23 from nexus20001/master

fix tests