Fixing an issue with the tests.