Change according to the test case