Update code so all tests succeed again