Oops, forgot to adjust tests.