Fix the tests I forgot I broke.