Fix for the tests.