Fix test case issue
Fix test case issue