Fix reference to mocks in tests.