Expanding functionality on mocking.