Мне кажется изменить класс проще, чем изменять все его вхождения в коде.