Fix test case regression