actually improve the tests, don't break them