Update test case accordingly