Unit Test case issue fixed