Fix test error
Merge pull request #7 from IainNZ/patch-1

Fix test error