Fix unit test breakage.