Fixed unit test breakage.