I should have learned to change tests when I change code a long time ago...