Instead modifying existent tests what do you think about writing new one for the new behavior?