Fix test
Merge pull request #2 from realmalex/master

Fix test