Fix issue with tests.