Fix failing test.
Merge pull request #1 from garymardell/fix

Fix failing test.