changed API for setting mocks