Fix test case issue