Added extra mocking.