Fixed the tests I broke last week.