Fix test case failure