Change test case accordingly