Replace stub with mocks