Fix test.
Merge pull request #5 from benalexau/master

Fix test.