Fixes broken test
Merge pull request #1564 from lukecampbell/tests

Fixes broken test