updated unit test case