unit test case fixed for step2