fix unit test breakage