Refactoring to list of return mocked objects