Fix unit tests I broke.