Modifying expectations in unit test to be less strict.