making the new tests pass