Patch up unit test I broke in pull request 7
Merge pull request #8 from negz/master

Patch up unit test I broke in pull request 7