Fix for this for tests.