Modify test case due to changes