Fix for the tests.
Merge pull request #89 from ChrisGrant/master

Fix for the tests.