The fact that I was lazy doesn't obligate you to cover the entire class. Just covering the new behavior would be fine.