modification on unit test case