Remove mocks from tests