Fix tests
Merge pull request #4 from retupmoca/master

Fix tests