Fix mock tests.