Mocking more classes