Update test case

Update test case