Fix test case breakage.