Unit test case issue fixed