Fixes broken test
Merge pull request #29 from lukecampbell/test-fix

Fixes broken test