Fix unit test that Alec Davis broke.