Change in a test to standardize mocking strategy