Fix test
Merge pull request #53 from Bronsa/master

Fix test