Fix your tests when you refactor a class, dummy!