Fix test case again