Fix test
Merge pull request #4 from papix/fix/test

Fix test