Fix for tests.